Update src/streamlit_app.py

src/streamlit_app.py

@@ -18,7 +18,7 @@ if not HF_TOKEN:
     st.stop()
 
 client = InferenceClient(
-    model="facebook/nllb-200-3.3B",
+    model="google/gemma-2b-it",   # ✅ FAST GEMMA MODEL
     token=HF_TOKEN
 )
 
@@ -68,35 +68,34 @@ def speech_to_text(audio_file):
     return text
 
 # ==============================
-# TRANSLATION FUNCTION (FIXED)
+# TRANSLATION FUNCTION (GEMMA)
 # ==============================
 def translate(text, target_lang):
 
-    # Detect language
     try:
         src_lang = detect(text)
     except:
         src_lang = "unknown"
 
-    # Retrieve memory
     memory = retrieve_memory(text)
 
-    # Prompt-based translation
     prompt = f"""
-You are a professional translator.
+You are an expert translator.
 
 Translate the following text into {target_lang}.
 
 Text:
 {text}
 
-Only return the translated text.
+IMPORTANT:
+- Only return translated text
+- Do not explain anything
 """
 
     try:
-        response = client.chat_completion(
+        response = client.text_generation(
             prompt,
-            max_tokens=200,
+            max_new_tokens=200,
             temperature=0.2
         )
 
@@ -105,7 +104,6 @@ Only return the translated text.
     except Exception as e:
         translated = f"❌ Translation failed: {str(e)}"
 
-    # Store memory
     store_memory(text, translated)
 
     return translated, src_lang, memory
@@ -121,15 +119,11 @@ tab1, tab2 = st.tabs(["📝 Text Input", "🎤 Voice Input"])
 
 input_text = ""
 
-# ------------------------------
 # TEXT INPUT
-# ------------------------------
 with tab1:
     input_text = st.text_area("Enter text")
 
-# ------------------------------
 # VOICE INPUT
-# ------------------------------
 with tab2:
     audio_file = st.file_uploader("Upload audio (wav/mp3)", type=["wav", "mp3"])
 
@@ -142,17 +136,13 @@ with tab2:
                 st.success("Recognized Text:")
                 st.write(input_text)
 
-# ------------------------------
 # TARGET LANGUAGE
-# ------------------------------
 target_lang = st.selectbox(
     "Target Language",
     ["English", "Tamil", "Hindi", "French", "Arabic", "Spanish", "German"]
 )
 
-# ------------------------------
-# TRANSLATE BUTTON
-# ------------------------------
+# TRANSLATE
 if st.button("Translate"):
     if not input_text.strip():
         st.warning("Please provide input text or audio")