update the api

app.py

@@ -4,13 +4,14 @@ from mtranslate import translate
 import requests
 
 HF_AUTH_TOKEN = os.environ.get("HF_AUTH_TOKEN")
-indochat_api = 'https://cahya-indonesian-whisperer.hf.space/api/indochat/v1'
-indochat_api_auth_token = os.getenv("INDOCHAT_API_AUTH_TOKEN", "")
+text_generator_api = 'https://cahya-indonesian-whisperer.hf.space/api/text-generator/v1'
+text_generator_api_auth_token = os.getenv("TEXT_GENERATOR_API_AUTH_TOKEN", "")
 
 def get_answer(user_input, decoding_method, num_beams, top_k, top_p, temperature, repetition_penalty, penalty_alpha):
     print(user_input, decoding_method, top_k, top_p, temperature, repetition_penalty, penalty_alpha)
-    headers = {'Authorization': 'Bearer ' + indochat_api_auth_token}
+    headers = {'Authorization': 'Bearer ' + text_generator_api_auth_token}
     data = {
+        "model_name": "bloomz-1b1-instruct",
         "text": user_input,
         "min_length": len(user_input) + 50,
         "max_length": 300,
@@ -23,7 +24,7 @@ def get_answer(user_input, decoding_method, num_beams, top_k, top_p, temperature
         "repetition_penalty": repetition_penalty,
         "penalty_alpha": penalty_alpha
     }
-    r = requests.post(indochat_api, headers=headers, data=data)
+    r = requests.post(text_generator_api, headers=headers, data=data)
     if r.status_code == 200:
         result = r.json()
         answer = result["generated_text"]
@@ -44,17 +45,15 @@ css = """
 
 with gr.Blocks(css=css) as demo:
     with gr.Row():
-        gr.Markdown("""## IndoChat
-        
-        A Prove of Concept of a multilingual Chatbot (in this case a bilingual, English and Indonesian), fine-tuned with 
-        multilingual instructions dataset. The base model is a GPT2-Medium (340M params) which was pretrained with 75GB 
-        of Indonesian and English dataset, where English part is only less than 1% of the whole dataset.
+        gr.Markdown("""## Bloomz-1b1-Instruct        
+        We fine-tuned the BigScience model Bloomz-1b1 with instructions dataset of following languages:
+        English, Indonesian, Malaysian, Vietnam, Hindi, Spanish, German, French, Russian, and Chinese.
         """)
     with gr.Row():
         with gr.Column():
             user_input = gr.inputs.Textbox(placeholder="",
-                                           label="Ask me something in Indonesian or English",
-                                           default="Bagaimana cara mendidik anak supaya tidak berbohong?")
+                                           label="Ask me something",
+                                           default="How old is the universe?")
             decoding_method = gr.inputs.Dropdown(["Beam Search", "Sampling", "Contrastive Search"],
                                                   default="Sampling", label="Decoding Method")
             num_beams = gr.inputs.Slider(label="Number of beams for beam search",