Update app.py

app.py

@@ -1,47 +1,29 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 
-# Initialize the inference client with the Mixtral model
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-def translate_text(text, target_language):
-    # Correctly format the prompt for the translation task
-    prompt = f"Translate the following text to {target_language}: {text}"
+def translate_text(input_text, target_language):
+    prompt = f"Translate the following text into {target_language}: {input_text}"
+    response = client.text_generation(prompt, parameters={"max_new_tokens": 256}, options={"wait_for_model": True})
     
-    # Correctly perform the model call for text generation
-    response = client.text_generation(
-        inputs=prompt,
-        parameters={"max_new_tokens": 100},
-        options={"wait_for_model": True}
-    )
-    
-    # Extract the translated text from the response
+    # Since the model's response includes the prompt, we extract only the translated text
+    # Assuming the translated text follows immediately after the prompt
     translated_text = response[0]['generated_text']
-    
-    # Depending on the model's response, you may need to clean the output
-    # For example, remove the input part from the response if necessary
-    translated_text = translated_text.split(prompt)[-1].strip()
-    
-    return translated_text
-
-languages = [
-    "French",
-    "Spanish",
-    "German",
-    "Italian",
-    "Portuguese",
-    # Add more languages as needed
-]
+    # Clean the response to display only the translated part
+    # This might need to be adjusted based on how the model includes the prompt in its response
+    clean_translation = translated_text[len(prompt):].strip()
+    return clean_translation
 
 iface = gr.Interface(
     fn=translate_text,
     inputs=[
-        gr.Textbox(label="Text to Translate", placeholder="Enter text here..."),
-        gr.Dropdown(label="Target Language", choices=languages)
+        gr.Textbox(label="Text to Translate", placeholder="Enter the text you want to translate here..."),
+        gr.Textbox(label="Target Language", placeholder="Enter the target language (e.g., French, Spanish)..."),
     ],
     outputs=gr.Textbox(label="Translated Text"),
     title="Simple Translator with Mixtral",
-    description="Translate text to various languages using the Mixtral model from Hugging Face."
+    description="Translate text to your specified language using the Mixtral model from Hugging Face."
 )
 
 iface.launch()