gdnartea commited on
Commit
dedd577
·
verified ·
1 Parent(s): e2294a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,5 +1,5 @@
1
  from transformers import pipeline
2
- from gradio import Interface, inputs, outputs
3
 
4
  # Initialize the text generation pipeline
5
  generator = pipeline('text-generation', model='microsoft/Phi-3-mini-4k-instruct-gguf')
@@ -10,10 +10,10 @@ def generate_text(prompt):
10
  return output[0]['generated_text']
11
 
12
  # Create a Gradio interface
13
- iface = Interface(
14
  fn=generate_text,
15
- inputs=inputs.Textbox(lines=5, placeholder="Enter your prompt here..."),
16
- outputs=outputs.Textbox()
17
  )
18
 
19
  # Launch the interface
 
1
  from transformers import pipeline
2
+ import gradio as gr
3
 
4
  # Initialize the text generation pipeline
5
  generator = pipeline('text-generation', model='microsoft/Phi-3-mini-4k-instruct-gguf')
 
10
  return output[0]['generated_text']
11
 
12
  # Create a Gradio interface
13
+ iface = gr.Interface(
14
  fn=generate_text,
15
+ inputs=gr.Textbox(lines=5, placeholder="Enter your prompt here..."),
16
+ outputs=gr.Textbox()
17
  )
18
 
19
  # Launch the interface