Update app.py
Browse files
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"])
|