Jezia commited on
Commit
d810c79
·
1 Parent(s): f558137

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,7 +1,10 @@
1
  import gradio as gr
2
  from gradio import inputs
3
  description = "Text generation with a miniature GPT"
4
- interface = gr.Interface.load("huggingface/keras-io/text-generation-miniature-gpt",
 
 
 
5
  title = "Text Generation with GPT-2",
6
  inputs = [
7
  gr.inputs.Textbox(lines=5, label="Text"),
@@ -9,4 +12,5 @@ interface = gr.Interface.load("huggingface/keras-io/text-generation-miniature-gp
9
  description=description,
10
  examples=[["A kid is playing with "]]
11
  )
 
12
  interface.launch()
 
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(model,
8
  title = "Text Generation with GPT-2",
9
  inputs = [
10
  gr.inputs.Textbox(lines=5, label="Text"),
 
12
  description=description,
13
  examples=[["A kid is playing with "]]
14
  )
15
+
16
  interface.launch()