DeMaking commited on
Commit
edad6c4
·
verified ·
1 Parent(s): bd8871f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ hf_hub_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
13
  login(token=hf_hub_token)
14
 
15
  # Load models
16
- hebrew_model = pipeline("text-generation", model="onlplab/alephbert-base")
17
  english_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
18
 
19
  @app.route("/ask", methods=["POST"])
 
13
  login(token=hf_hub_token)
14
 
15
  # Load models
16
+ hebrew_model = pipeline("text-generation", model="onlplab/alephbert-base", is_decoder=True)
17
  english_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
18
 
19
  @app.route("/ask", methods=["POST"])