Spaces:
Running
Running
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=[
|