Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=[
|