Update app.py

app.py

@@ -23,7 +23,7 @@ login(token=HF_HUB_TOKEN)
 # # # Load Hebrew and English text generation models
 # hebrew_generator = pipeline("text-generation", model="onlplab/alephbert-base")
 # english_generator = pipeline("text-generation", model="vicgalle/gpt2-open-instruct-v1")
-english_generator = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct")
+lang_generator = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct")
 
 
 # # hebrew_generator = pipeline("text-generation", model="Norod78/hebrew-gpt_neo-small")
@@ -72,17 +72,19 @@ def detect_language(user_input):
 def generate_response(text):
     language = detect_language(text)
     print(f"Detected language: {language}, ", f"current time: {current_time_gmt()}")
-
+        
     if language == "hebrew":
-        hebrew_generator = pipeline("text-generation", model="onlplab/alephbert-base")
-        output = hebrew_generator(text, max_length=100, truncation=True)
+        #hebrew_generator = pipeline("text-generation", model="onlplab/alephbert-base")
+        # output = hebrew_generator(text, max_length=100, truncation=True)
+        output = lang_generator(text, max_length=100, truncation=True)
         print(f"Hebrew model output: {output}, ", f"current time: {current_time_gmt()}")  # Debugging
         return output[0]["generated_text"]
     
     elif language == "english":
         #english_generator = pipeline("text-generation", model="mistralai/Mistral-Nemo-Instruct-2407", max_new_tokens=128)
         # english_generator = pipeline("text-generation", model="distilgpt2")
-        output = english_generator(text, max_length=50, truncation=True)
+        #output = english_generator(text, max_length=50, truncation=True)
+        output = lang_generator(text, max_length=50, truncation=True)
         print(f"English model output: {output}, ", f"current time: {current_time_gmt()}")  # Debugging
         return output[0]["generated_text"]