ArunAIML commited on
Commit
9df41bd
·
1 Parent(s): 30c9c77

added torch

Browse files
Files changed (2) hide show
  1. app.py +14 -4
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,7 +1,17 @@
 
1
  import gradio as gr
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
5
 
6
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- iface.launch()
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
  import gradio as gr
3
 
4
+ model_id = "gpt2-large"
 
5
 
6
+ pipe = pipeline("text-generation", model=model_id)
7
+
8
+ def gpt(prompt):
9
+ return pipe(prompt, top_k=top_k, penalty_alpha=penalty_alpha)[0]["generated_text"]
10
+
11
+ gr.Interface(
12
+ explore_gpt_model,
13
+ ["text",gr.Slider(minimum=0, maximum=50, step=1,label="Top_k"),gr.Slider(minimum=0.1, maximum=1.0,label="penalty_alpha")],
14
+ "text",
15
+ title= "Arun's GPT chatbot",
16
+ description = "This is Arun's experimental GPT interface exposing gpt2-large, feel free to experiment"
17
+ ).launch()
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
  transformers
2
  gradio
 
 
1
  transformers
2
  gradio
3
+ torch