Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,17 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
3 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
# بارگذاری مدل فارسی
|
5 |
model_name = "HooshvareLab/gpt2-fa"
|
6 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
@@ -18,6 +29,11 @@ def student_bot(question):
|
|
18 |
پاسخ: حدود یک هفته بعد از شروع ترم تحصیلی است.
|
19 |
""" + "\n".join(chat_history[-3:]) + "\nپاسخ:"
|
20 |
|
|
|
|
|
|
|
|
|
|
|
21 |
inputs = tokenizer(prompt, return_tensors="pt")
|
22 |
outputs = model.generate(
|
23 |
**inputs,
|
@@ -32,9 +48,11 @@ def student_bot(question):
|
|
32 |
# رابط Gradio
|
33 |
gr.Interface(
|
34 |
fn=student_bot,
|
35 |
-
inputs=gr.Textbox(label="سوال
|
36 |
-
outputs=gr.Textbox(label="پاسخ"),
|
37 |
-
title="🎓 ایجنت راهنمای دانشجویان",
|
38 |
-
description="پاسخ به سوالات با کمک مدل
|
|
|
39 |
).launch()
|
40 |
|
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
3 |
|
4 |
+
|
5 |
+
# بارگذاری پرسش و پاسخهای FAQ
|
6 |
+
faq_dict = {}
|
7 |
+
with open("faq.txt", encoding="utf-8") as f:
|
8 |
+
for line in f:
|
9 |
+
if ":" in line:
|
10 |
+
q, a = line.strip().split(":", 1)
|
11 |
+
faq_dict[q.strip()] = a.strip()
|
12 |
+
|
13 |
+
|
14 |
+
|
15 |
# بارگذاری مدل فارسی
|
16 |
model_name = "HooshvareLab/gpt2-fa"
|
17 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
29 |
پاسخ: حدود یک هفته بعد از شروع ترم تحصیلی است.
|
30 |
""" + "\n".join(chat_history[-3:]) + "\nپاسخ:"
|
31 |
|
32 |
+
for key in faq_dict:
|
33 |
+
if key in question:
|
34 |
+
return faq_dict[key]
|
35 |
+
|
36 |
+
|
37 |
inputs = tokenizer(prompt, return_tensors="pt")
|
38 |
outputs = model.generate(
|
39 |
**inputs,
|
|
|
48 |
# رابط Gradio
|
49 |
gr.Interface(
|
50 |
fn=student_bot,
|
51 |
+
inputs=gr.Textbox(label="❓ سوال خود را وارد کنید"),
|
52 |
+
outputs=gr.Textbox(label="✅ پاسخ هوشمند"),
|
53 |
+
title="🎓 ایجنت راهنمای دانشجویان دانشگاه",
|
54 |
+
description="پاسخ به سوالات پرتکرار آموزشی با کمک مدل FarsiGPT و پایگاه دانش",
|
55 |
+
theme="compact"
|
56 |
).launch()
|
57 |
|
58 |
+
|