DeMaking commited on
Commit
b04f9bb
·
verified ·
1 Parent(s): 6835f36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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
- # english_generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
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."