Spaces:

guychuk
/

vit-small-ps

Runtime error

guychuk commited on May 14

Commit

662abb8

verified ·

1 Parent(s): 54154aa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,6 @@
 import gradio as gr
 from transformers import pipeline
-from optimum.pipelines import onnx_pipeline
 # Load Models
 clean_pipe = pipeline("image-classification", model="WinKawaks/vit-small-patch16-224")

 import gradio as gr
 from transformers import pipeline
+from optimum.pipelines import pipeline as onnx_pipeline
 # Load Models
 clean_pipe = pipeline("image-classification", model="WinKawaks/vit-small-patch16-224")