syurek commited on
Commit
79bdebe
·
verified ·
1 Parent(s): c5d4a8e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # Sohbet modelini yükle (DialoGPT)
5
- chatbot = pipeline("conversational", model="microsoft/DialoGPT-medium")
6
 
7
  # JSON dosyasındaki verilerden alınan sorulara cevap veren fonksiyon
8
  def answer_question(file, question):
@@ -23,8 +23,8 @@ def answer_question(file, question):
23
  return f"JSON verisinde {people_count} kişi var."
24
 
25
  # Eğer farklı bir şey sorulursa, sohbet modelini kullanarak yanıt ver
26
- response = chatbot(question)
27
- return str(response)
28
 
29
  except Exception as e:
30
  return f"Hata: {e}"
 
2
  from transformers import pipeline
3
 
4
  # Sohbet modelini yükle (DialoGPT)
5
+ chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
6
 
7
  # JSON dosyasındaki verilerden alınan sorulara cevap veren fonksiyon
8
  def answer_question(file, question):
 
23
  return f"JSON verisinde {people_count} kişi var."
24
 
25
  # Eğer farklı bir şey sorulursa, sohbet modelini kullanarak yanıt ver
26
+ response = chatbot(question, max_length=1000, num_return_sequences=1)[0]['generated_text']
27
+ return response
28
 
29
  except Exception as e:
30
  return f"Hata: {e}"