Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,8 @@ embedding_tokenizer = AutoTokenizer.from_pretrained(embedding_model_name)
|
|
| 12 |
embedding_model = AutoModel.from_pretrained(embedding_model_name)
|
| 13 |
|
| 14 |
# 🔹 مدل تولید (برای پاسخ جدید)
|
| 15 |
-
gen_model_name = "HooshvareLab/PersianMind"
|
|
|
|
| 16 |
gen_tokenizer = AutoTokenizer.from_pretrained(gen_model_name)
|
| 17 |
gen_model = AutoModelForCausalLM.from_pretrained(gen_model_name)
|
| 18 |
|
|
|
|
| 12 |
embedding_model = AutoModel.from_pretrained(embedding_model_name)
|
| 13 |
|
| 14 |
# 🔹 مدل تولید (برای پاسخ جدید)
|
| 15 |
+
#gen_model_name = "HooshvareLab/PersianMind"
|
| 16 |
+
gen_model_name = "universitytehran/PersianMind-v1.0"
|
| 17 |
gen_tokenizer = AutoTokenizer.from_pretrained(gen_model_name)
|
| 18 |
gen_model = AutoModelForCausalLM.from_pretrained(gen_model_name)
|
| 19 |
|