Update app.py

app.py

@@ -5,16 +5,16 @@ from langdetect import detect
 app = Flask(__name__)
 
 # Load models
-hebrew_model = pipeline("text-generation", model="onlplab/alephbertgpt") 
+hebrew_model = pipeline("text-generation", model="onlplab/alephbert-base")
 english_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct")
 
-@app.route("/generate", methods=["POST"])
-def generate():
+@app.route("/ask", methods=["POST"])
+def ask():
     data = request.json
-    text = data.get("text", "")
+    user_input = data.get("text", "")
 
     # Detect language 
-    language = detect(text)
+    language = detect(user_input)
 
     if language == 'he':
         model = hebrew_model
@@ -24,7 +24,7 @@ def generate():
         print("Decision Making Helper BOT currently supports Hebrew and English Languages")
 
     # Create an answer from the model
-    response = model(text, max_length=100, do_sample=True)
+    response = model(user_input, max_length=100, do_sample=True)
     return jsonify({"response": response[0]['generated_text']})
 
 if __name__ == "__main__":