DeMaking commited on
Commit
73bc039
·
verified ·
1 Parent(s): 489d405

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ login(token=hf_hub_token)
16
 
17
 
18
  # Load Hebrew text generation model
19
- hebrew_generator = pipeline("text-generation", model="google/mt5-small", tokenizer="google/mt5-small")
20
 
21
  # Load English text generation model
22
  english_generator = pipeline("text-generation", model="distilgpt2")
 
16
 
17
 
18
  # Load Hebrew text generation model
19
+ hebrew_generator = pipeline("text-generation", model="Norod78/hebrew-gpt_neo-small")
20
 
21
  # Load English text generation model
22
  english_generator = pipeline("text-generation", model="distilgpt2")