Younes13 commited on
Commit
7be27f8
·
verified ·
1 Parent(s): e389e19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -4
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="پاسخ به سوالات با کمک مدل زبانی فارسی FarsiGPT"
 
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
+