diginoron commited on
Commit
3d6570a
·
verified ·
1 Parent(s): b6fdddd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -2,6 +2,7 @@ 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")
@@ -30,13 +31,18 @@ def generate_topics(field, major, keywords, audience, level):
30
  )
31
 
32
  try:
33
- # فراخوانی مدل: فقط پرامپت به‌صورت موقعیتی
34
- result = hf_client.text_generation(
35
- prompt,
 
 
 
36
  max_new_tokens=512,
37
  temperature=0.7
38
  )
39
- english_output = result.generated_text.strip()
 
 
40
 
41
  # ترجمه به فارسی
42
  try:
 
2
  import os
3
  from deep_translator import GoogleTranslator
4
  from huggingface_hub import InferenceClient
5
+ from transformers import Conversation
6
 
7
  # دریافت توکن از محیط (در Hugging Face Secrets تنظیم شود)
8
  HF_TOKEN = os.environ.get("HUGGINGFACE_API_TOKEN")
 
31
  )
32
 
33
  try:
34
+ # آماده‌سازی یک دیالوگ
35
+ conv = Conversation(prompt)
36
+
37
+ # فراخوانی مدل به‌صورت conversational
38
+ hf_client.conversational(
39
+ conv,
40
  max_new_tokens=512,
41
  temperature=0.7
42
  )
43
+
44
+ # دریافت آخرین پاسخ مدل
45
+ english_output = conv.generated_responses[-1].strip()
46
 
47
  # ترجمه به فارسی
48
  try: