Update app.py

app.py

@@ -20,9 +20,12 @@ if not HF_HUB_TOKEN:
 login(token=HF_HUB_TOKEN)
 
 
-# Load Hebrew and English text generation models
-hebrew_generator = pipeline("text-generation", model="Norod78/hebrew-gpt_neo-small")
-english_generator = pipeline("text-generation", model="distilgpt2")
+# # Load Hebrew and English text generation models
+hebrew_generator = pipeline("text-generation", model="onlplab/alephbert-base")
+english_generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
+
+# hebrew_generator = pipeline("text-generation", model="Norod78/hebrew-gpt_neo-small")
+# english_generator = pipeline("text-generation", model="distilgpt2")
 
 
 # Function to detect language
@@ -122,19 +125,23 @@ async def generate_text(request: Request):
     try:
         data = await request.json()
         if not data or "text" not in data:
-            logging.error("Received an empty or invalid request")
+            logging.error("Invalid request received")
             return {"error": "Invalid request. Please send JSON with a 'text' field."}
 
         text = data["text"].strip()
         if not text:
             return {"error": "No text provided"}
 
+        print(f"Received text: {text}")  # Debugging
+        
         response = generate_response(text)
+        print(f"Generated response: {response}")  # Debugging
         return {"response": response}
 
     except Exception as e:
         logging.error(f"Error processing request: {e}")
-        return {"error": "Invalid request. Please send JSON with a 'text' field."}
+        return {"error": "An unexpected error occurred."}
+