Spaces:

miittnnss
/

Pet-Classifier

Sleeping

miittnnss commited on Jan 22, 2024

Commit

29e65a9

verified ·

1 Parent(s): 1b2bb40

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,5 @@
 from transformers import pipeline
-import torch
 def select_model(model_name):
     return pipeline("image-classification", model=model_name)

 from transformers import pipeline
+import gradio as gr
 def select_model(model_name):
     return pipeline("image-classification", model=model_name)