Update app.py

app.py

@@ -6,14 +6,8 @@ client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 def translate_text(input_text, target_language):
     prompt = f"Translate the following text into {target_language}: {input_text}"
     response = client.text_generation(prompt)
-    
-    # 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']
-    # 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
+    translated_text = response['generated_text'] if 'generated_text' in response else "Translation error or model response format has changed."
+    return translated_text
 
 iface = gr.Interface(
     fn=translate_text,
@@ -26,4 +20,4 @@ iface = gr.Interface(
     description="Translate text to your specified language using the Mixtral model from Hugging Face."
 )
 
-iface.launch()
+iface.launch()