Update app.py

app.py

@@ -22,7 +22,7 @@ translator_client = Client("Chatboong/Gemini_Translator")
 
 
 def translate_zomi(text: str):
-    msg = f"Translate Zomi to English: '{text}'"
+    msg = f"Translate Zomi to English, if it is English translate it to Zomi: '{text}'"
 
     stream = translator_client.predict(
         message=msg,
@@ -36,7 +36,7 @@ def translate_zomi(text: str):
         output += str(chunk)
 
         # Remove prefix 
-        prefix = "Translate Zomi to English:" 
+        prefix = "Translate Zomi to English, if it is English translate it to Zomi: " 
         if output.startswith(prefix): 
           output = output[len(prefix):].strip() 
           # Remove surrounding quotes