arielleharris commited on
Commit
f59f402
·
verified ·
1 Parent(s): 9e91338

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  from transformers import pipeline
 
2
 
3
  # Load a text-generation model
4
  chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
 
1
  from transformers import pipeline
2
+ import gradio as gr # Import Gradio for the interface
3
 
4
  # Load a text-generation model
5
  chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")