Jezia commited on
Commit
f306d19
·
1 Parent(s): 88fd916

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -15
app.py CHANGED
@@ -1,16 +1,29 @@
1
  import gradio as gr
2
- from gradio import inputs
3
- description = "Text generation with a miniature GPT"
4
- from huggingface_hub import from_pretrained_keras
5
- model = from_pretrained_keras("huggingface/keras-io/text-generation-miniature-gpt")
6
-
7
- interface = gr.Interface.load("huggingface/pranavpsv/gpt2-genre-story-generator",
8
- title = "Text Generation with GPT-2",
9
- inputs = [
10
- gr.inputs.Textbox(lines=5, label="Text"),
11
- ],
12
- description=description,
13
- examples=[["A kid is playing with "]]
14
- )
15
-
16
- interface.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ from gradio import mix
3
+
4
+ title = "Miniature"
5
+ description = "Gradio Demo for a miniature with GPT. To use it, simply add your text, or click one of the examples to load them. Read more at the links below."
6
+
7
+
8
+ examples = [
9
+ ['A kid is playing with']
10
+ ]
11
+
12
+ io = gr.Interface.load("models/sberbank-ai/rugpt3large_based_on_gpt2")
13
+
14
+
15
+
16
+ def inference(text):
17
+
18
+ return io(text)
19
+
20
+
21
+
22
+ gr.Interface(
23
+ inference,
24
+ [gr.inputs.Textbox(label="Input")],
25
+ gr.outputs.Textbox(label="Output"),
26
+ examples=examples,
27
+ # article=article,
28
+ title=title,
29
+ description=description).launch(enable_queue=True, cache_examples=True)