Spaces:

turtlegraphics
/

image-recognition

Sleeping

turtlegraphics commited on Mar 26

Commit

1d99f23

verified ·

1 Parent(s): 4280a69

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ description = "Place to try various models"
 def classify(image):
     inputs = feature_extractor(images=image, return_tensors="pt")
     outputs = model(**inputs)
-    return "dog"
 demo = gr.Interface(fn=classify, inputs="image", outputs="text")

 def classify(image):
     inputs = feature_extractor(images=image, return_tensors="pt")
     outputs = model(**inputs)
+    return str(outputs)
 demo = gr.Interface(fn=classify, inputs="image", outputs="text")