Update app.py

app.py

@@ -8,19 +8,18 @@ from huggingface_hub import login
 # Initialize Flask app
 app = Flask(__name__)
 
-# Gets the Token from secrect
+# Gets the Token from secrects and Login if exists
 hf_hub_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
-
-# Logging in
+if not hf_hub_token:
+    raise ValueError("Missing Hugging Face API token. Please set HUGGINGFACEHUB_API_TOKEN in environment variables.")
 login(token=hf_hub_token)
 
-# Load Hebrew models (classification + text generation)
-hebrew_classifier = pipeline("text-classification", model="onlplab/alephbert-base")
-hebrew_generator = pipeline("text2text-generation", model="google/mt5-small")
 
-# Load English models
-english_classifier = pipeline("text-classification", model="mistralai/Mistral-7B-Instruct-v0.3")
-english_generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
+# Load Hebrew text generation model
+hebrew_generator = pipeline("text-generation", model="google/mt5-small", tokenizer="google/mt5-small")
+
+# Load English text generation model
+english_generator = pipeline("text-generation", model="distilgpt2")
 
 # Function to detect language
 def detect_language(user_input):
@@ -42,13 +41,11 @@ def generate_response(text):
     language = detect_language(text)
 
     if language == "hebrew":
-        classification = hebrew_classifier(text)
         response = hebrew_generator(text, max_length=100)[0]["generated_text"]
     elif language == "english":
-        classification = english_classifier(text)
         response = english_generator(text, max_length=100)[0]["generated_text"]
     else:
-        response = "❌ Sorry, I only support Hebrew and English."
+        response = "Sorry, I only support Hebrew and English."
 
     return response
 
@@ -69,7 +66,7 @@ def ask():
 # Root endpoint
 @app.route("/")
 def home():
-    return "Decision Helper Bot API is running!"
+    return "Decision Making Helper Bot API is running!"
 
 if __name__ == "__main__":
     logging.basicConfig(level=logging.INFO)