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."
|