Skier8402 commited on
Commit
304d993
·
1 Parent(s): eb53c79

Update app.py

Browse files

Add inference step

Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -7,3 +7,8 @@ model = pipeline("text-generation")
7
  def predict(prompt):
8
  completion = model(prompt)[0]["generated_text"]
9
  return completion
 
 
 
 
 
 
7
  def predict(prompt):
8
  completion = model(prompt)[0]["generated_text"]
9
  return completion
10
+
11
+
12
+ import gradio as gr
13
+
14
+ gr.Interface(fn=predict, inputs="text", outputs="text").launch()