TASOMM commited on
Commit
ff5021e
·
verified ·
1 Parent(s): 8cfc12f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,12 +1,12 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- pipe = pipeline("text-generation", model="tiiuae/falcon-7b-instruct")
 
5
 
6
  def chat_fn(message, history):
7
- response = pipe(message, max_length=200, do_sample=True, temperature=0.7)
8
  return response[0]["generated_text"]
9
 
10
  demo = gr.ChatInterface(chat_fn)
11
-
12
  demo.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Küçük model (çalışması garanti)
5
+ pipe = pipeline("text2text-generation", model="google/flan-t5-small")
6
 
7
  def chat_fn(message, history):
8
+ response = pipe(message, max_length=200)
9
  return response[0]["generated_text"]
10
 
11
  demo = gr.ChatInterface(chat_fn)
 
12
  demo.launch()