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"])
|