someone commited on
Commit
b7abecf
1 Parent(s): 85df26c
Files changed (2) hide show
  1. app.py +11 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline, set_seed
3
+
4
+ def greet(name):
5
+ generator = pipeline('text-generation', model='gpt2')
6
+ set_seed(42)
7
+ res = generator(name, max_length=10, num_return_sequences=5)
8
+ return str(res)
9
+
10
+ iface = gr.Interface(fn=greet, inputs="text", outputs="text")
11
+ iface.launch()
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ torch
2
+ transformers
3
+