diginoron commited on
Commit
a95f6dc
·
verified ·
1 Parent(s): ae1598f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -3,10 +3,10 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
3
  from deep_translator import GoogleTranslator
4
  import torch
5
 
6
- # مدل پیشنهادی پایدار
7
  model_id = "google/gemma-7b-it"
8
 
9
- # بارگذاری توکنایزر و مدل
10
  tokenizer = AutoTokenizer.from_pretrained(model_id)
11
  model = AutoModelForCausalLM.from_pretrained(
12
  model_id,
@@ -15,7 +15,7 @@ model = AutoModelForCausalLM.from_pretrained(
15
  )
16
  model.eval()
17
 
18
- # تابع تولید موضوع پایان‌نامه
19
  def generate_topics(field, major, keywords, audience, level):
20
  prompt = f"""<bos>[INST]Suggest 3 academic thesis topics based on the following:
21
  Field: {field}
@@ -37,14 +37,15 @@ Level: {level}[/INST]"""
37
 
38
  html_output = (
39
  "<div dir='rtl' style='text-align: right; font-family: Tahoma, sans-serif; "
40
- "font-size: 16px; line-height: 1.8; min-height: 250px;'>"
 
41
  f"{translated_output_html}"
42
  "<br><br>📢 برای مشاوره و راهنمایی تخصصی با گروه مشاوره کاسپین تماس بگیرید:<br>"
43
  "<strong>021-88252497</strong></div>"
44
  )
45
  return html_output
46
 
47
- # رابط Gradio
48
  iface = gr.Interface(
49
  fn=generate_topics,
50
  inputs=[
 
3
  from deep_translator import GoogleTranslator
4
  import torch
5
 
6
+ # انتخاب مدل پایدار
7
  model_id = "google/gemma-7b-it"
8
 
9
+ # بارگذاری مدل و توکنایزر
10
  tokenizer = AutoTokenizer.from_pretrained(model_id)
11
  model = AutoModelForCausalLM.from_pretrained(
12
  model_id,
 
15
  )
16
  model.eval()
17
 
18
+ # تابع اصلی پیشنهاد موضوع پایان‌نامه
19
  def generate_topics(field, major, keywords, audience, level):
20
  prompt = f"""<bos>[INST]Suggest 3 academic thesis topics based on the following:
21
  Field: {field}
 
37
 
38
  html_output = (
39
  "<div dir='rtl' style='text-align: right; font-family: Tahoma, sans-serif; "
40
+ "font-size: 16px; line-height: 1.8; min-height: 250px; max-height: 400px; "
41
+ "overflow-y: auto; border: 1px solid #ccc; padding: 10px; background-color: #fdfdfd;'>"
42
  f"{translated_output_html}"
43
  "<br><br>📢 برای مشاوره و راهنمایی تخصصی با گروه مشاوره کاسپین تماس بگیرید:<br>"
44
  "<strong>021-88252497</strong></div>"
45
  )
46
  return html_output
47
 
48
+ # رابط کاربری Gradio
49
  iface = gr.Interface(
50
  fn=generate_topics,
51
  inputs=[