Spaces:

stevhliu
/

gradio-pipeline-demo

Sleeping

stevhliu HF staff commited on Aug 29

Commit

297b974

•

1 Parent(s): ea8eb6d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,5 @@
 from transformers import pipeline
 import gradio as gr
-pipeline = pipeline("text-generation", model="openai-community/gpt2")
 gr.Interface.from_pipeline(pipeline).launch()

 from transformers import pipeline
 import gradio as gr
+pipeline = pipeline("image-classification", model="google/vit-base-patch16-224")
 gr.Interface.from_pipeline(pipeline).launch()