Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ login(token=HF_HUB_TOKEN)
|
|
22 |
|
23 |
# # # Load Hebrew and English text generation models
|
24 |
# hebrew_generator = pipeline("text-generation", model="onlplab/alephbert-base")
|
25 |
-
|
26 |
|
27 |
# # hebrew_generator = pipeline("text-generation", model="Norod78/hebrew-gpt_neo-small")
|
28 |
# # english_generator = pipeline("text-generation", model="distilgpt2")
|
@@ -69,17 +69,19 @@ def detect_language(user_input):
|
|
69 |
|
70 |
def generate_response(text):
|
71 |
language = detect_language(text)
|
72 |
-
print(f"Detected language: {language}")
|
73 |
|
74 |
if language == "hebrew":
|
75 |
hebrew_generator = pipeline("text-generation", model="onlplab/alephbert-base")
|
76 |
output = hebrew_generator(text, max_length=100, truncation=True)
|
|
|
77 |
return output[0]["generated_text"]
|
78 |
|
79 |
elif language == "english":
|
80 |
-
english_generator = pipeline("text-generation", model="mistralai/Mistral-Nemo-Instruct-2407", max_new_tokens=128)
|
81 |
# english_generator = pipeline("text-generation", model="distilgpt2")
|
82 |
output = english_generator(text, max_length=50, truncation=True)
|
|
|
83 |
return output[0]["generated_text"]
|
84 |
|
85 |
return "Sorry, I only support Hebrew and English."
|
|
|
22 |
|
23 |
# # # Load Hebrew and English text generation models
|
24 |
# hebrew_generator = pipeline("text-generation", model="onlplab/alephbert-base")
|
25 |
+
english_generator = pipeline("text-generation", model="mistralai/Mistral-Nemo-Instruct-2407", max_new_tokens=128)
|
26 |
|
27 |
# # hebrew_generator = pipeline("text-generation", model="Norod78/hebrew-gpt_neo-small")
|
28 |
# # english_generator = pipeline("text-generation", model="distilgpt2")
|
|
|
69 |
|
70 |
def generate_response(text):
|
71 |
language = detect_language(text)
|
72 |
+
print(f"Detected language: {language}", f"current time: {current_time_gmt()})
|
73 |
|
74 |
if language == "hebrew":
|
75 |
hebrew_generator = pipeline("text-generation", model="onlplab/alephbert-base")
|
76 |
output = hebrew_generator(text, max_length=100, truncation=True)
|
77 |
+
print(f"Hebrew model output: {output}, ", f"current time: {current_time_gmt()}") # Debugging
|
78 |
return output[0]["generated_text"]
|
79 |
|
80 |
elif language == "english":
|
81 |
+
#english_generator = pipeline("text-generation", model="mistralai/Mistral-Nemo-Instruct-2407", max_new_tokens=128)
|
82 |
# english_generator = pipeline("text-generation", model="distilgpt2")
|
83 |
output = english_generator(text, max_length=50, truncation=True)
|
84 |
+
print(f"English model output: {output}, ", f"current time: {current_time_gmt()}") # Debugging
|
85 |
return output[0]["generated_text"]
|
86 |
|
87 |
return "Sorry, I only support Hebrew and English."
|