Update app.py

app.py

@@ -23,6 +23,26 @@ def generate(prompt, system_prompt, max_new_tokens):
     formatted_prompt = f" [INST] {system_prompt}, {prompt} [/INST]"
     output = ""
 
+    for response in client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False):
+        output += response.token.text
+
+    return translate(output)
+
+def translate(input):
+    generate_kwargs = dict(
+        temperature=0.7,
+        max_new_tokens=512,
+        top_p=0.95,
+        repetition_penalty=1.0,
+        do_sample=True,
+        seed=42,
+    )
+
+    system_prompt = "Traduza todo e qualquer texto que o usuário entregar para português."
+
+    formatted_prompt = f" [INST] {system_prompt}, {prompt} [/INST]"
+    output = ""
+
     for response in client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False):
         output += response.token.text
 
@@ -51,7 +71,7 @@ def caption(img, min_len, max_len):
 
 iface= gr.Interface(fn=caption, 
                      title='Blip Image Captioning Large', 
-                     description="[Salesforce/blip-image-captioning-large](https://huggingface.co/Salesforce/blip-image-captioning-large)",
+                     description="._.",
                      inputs=[gr.Image(type='filepath', label='Image'), gr.Slider(label='Minimum Length', minimum=1, maximum=1000, value=30), gr.Slider(label='Maximum Length', minimum=1, maximum=1000, value=100)], 
                      outputs=gr.Textbox(label='Caption'),
                      theme = gr.themes.Base(primary_hue="teal",secondary_hue="teal",neutral_hue="slate"),)