Ajay12345678980 commited on
Commit
b554450
·
verified ·
1 Parent(s): e26be7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -24,6 +24,14 @@ def predict(text):
24
  # If no match is found, return the full prediction
25
  return prediction.strip()
26
 
 
 
 
 
 
 
 
 
27
 
28
  # Launch the Gradio app
29
  if __name__ == "__main__":
 
24
  # If no match is found, return the full prediction
25
  return prediction.strip()
26
 
27
+ interface = gr.Interface(
28
+ fn=predict,
29
+ inputs="text",
30
+ outputs="text",
31
+ title="GPT-2 Text Generation",
32
+ description="Enter some text and see what the model generates!"
33
+ )
34
+
35
 
36
  # Launch the Gradio app
37
  if __name__ == "__main__":