diginoron commited on
Commit
097fb1d
·
verified ·
1 Parent(s): d0b4a83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -15
app.py CHANGED
@@ -1,20 +1,15 @@
1
  import gradio as gr
2
  import os
3
  from deep_translator import GoogleTranslator
4
- from huggingface_hub import InferenceApi
5
 
6
- # دریافت کلید از محیط (در Hugging Face از Secrets استفاده کنید)
7
  HF_TOKEN = os.environ.get("HUGGINGFACE_API_TOKEN")
8
  if not HF_TOKEN:
9
  raise RuntimeError("Missing HUGGINGFACE_API_TOKEN secret")
10
 
11
- # ایجاد کلاینت برای تماس با مدل DeepSeek
12
- hf_api = InferenceApi(
13
- repo_id="deepseek-ai/DeepSeek-Prover-V2-671B",
14
- token=HF_TOKEN
15
- )
16
-
17
- # تابع اصلی پیشنهاد موضوع پایان‌نامه
18
 
19
  def generate_topics(field, major, keywords, audience, level):
20
  # اعتبارسنجی ورودی‌ها
@@ -32,13 +27,14 @@ def generate_topics(field, major, keywords, audience, level):
32
  )
33
 
34
  try:
35
- # فراخوانی مدل DeepSeek
36
- response = hf_api(
 
37
  inputs=prompt,
38
- parameters={"max_new_tokens": 512, "temperature": 0.7}
 
39
  )
40
- # استخراج متن پاسخ (در صورت نیاز به تنظیم بر اساس فرمت خروجی)
41
- english_output = response[0]['generated_text'].strip() if isinstance(response, list) else response.strip()
42
 
43
  # ترجمه به فارسی
44
  try:
@@ -46,7 +42,7 @@ def generate_topics(field, major, keywords, audience, level):
46
  except Exception:
47
  translated_output = english_output
48
 
49
- # قالب‌بندی خروجی HTML
50
  translated_output_html = "<ol>" + \
51
  "".join(f"<li>{line}</li>" for line in translated_output.split("\n") if line) + \
52
  "</ol>"
@@ -96,4 +92,5 @@ iface = gr.Interface(
96
  )
97
 
98
  if __name__ == "__main__":
 
99
  iface.launch()
 
1
  import gradio as gr
2
  import os
3
  from deep_translator import GoogleTranslator
4
+ from huggingface_hub import InferenceClient
5
 
6
+ # دریافت توکن از محیط (در Hugging Face Secrets تنظیم شود)
7
  HF_TOKEN = os.environ.get("HUGGINGFACE_API_TOKEN")
8
  if not HF_TOKEN:
9
  raise RuntimeError("Missing HUGGINGFACE_API_TOKEN secret")
10
 
11
+ # ایجاد کلاینت جدید
12
+ hf_client = InferenceClient(token=HF_TOKEN)
 
 
 
 
 
13
 
14
  def generate_topics(field, major, keywords, audience, level):
15
  # اعتبارسنجی ورودی‌ها
 
27
  )
28
 
29
  try:
30
+ # فراخوانی مدل DeepSeek با InferenceClient
31
+ result = hf_client.text_generation(
32
+ model="deepseek-ai/DeepSeek-Prover-V2-671B",
33
  inputs=prompt,
34
+ max_new_tokens=512,
35
+ temperature=0.7
36
  )
37
+ english_output = result.generated_text.strip()
 
38
 
39
  # ترجمه به فارسی
40
  try:
 
42
  except Exception:
43
  translated_output = english_output
44
 
45
+ # قالب‌بندی HTML به‌صورت لیست مرتب
46
  translated_output_html = "<ol>" + \
47
  "".join(f"<li>{line}</li>" for line in translated_output.split("\n") if line) + \
48
  "</ol>"
 
92
  )
93
 
94
  if __name__ == "__main__":
95
+ # اگر نیاز به لینک عمومی دارید، share=True اضافه کنید
96
  iface.launch()