Spaces:

Adhir
/

test-medium

Sleeping

Adhir commited on Apr 19

Commit

b9b1c0f

•

1 Parent(s): 0c8f4ea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import gradio as gr
 from transformers import pipeline
 #pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
-pipeline = pipeline(task="image-classification", model="hg2001/autotrain-animals-vs-humans2-37846100283")
 def predict(image):

 from transformers import pipeline
 #pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
+#pipeline = pipeline(task="image-classification", model="hg2001/autotrain-animals-vs-humans2-37846100283")
+pipeline = pipeline("image-classification", model="lazyturtl/roomclassifier")
 def predict(image):