ghosttech commited on
Commit
b016bd9
·
verified ·
1 Parent(s): 25be645

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,5 +11,6 @@ def generate_response(prompt):
11
  output = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7)
12
  return output[0]['generated_text']
13
 
14
- demo = gr.Interface(fn=generate_response, inputs="text", outputs="text")
 
15
  demo.launch()
 
11
  output = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7)
12
  return output[0]['generated_text']
13
 
14
+ # api_name parametresi ile dışarıdan gelen /predict isteklerinin generate_response fonksiyonunu tetiklemesini sağlıyoruz
15
+ demo = gr.Interface(fn=generate_response, inputs="text", outputs="text", api_name="predict")
16
  demo.launch()