Adhir commited on
Commit
0c8f4ea
1 Parent(s): 92a8827

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,7 +3,10 @@
3
  import gradio as gr
4
  from transformers import pipeline
5
 
6
- pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
 
 
7
 
8
  def predict(image):
9
  predictions = pipeline(image)
 
3
  import gradio as gr
4
  from transformers import pipeline
5
 
6
+ #pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
7
+ pipeline = pipeline(task="image-classification", model="hg2001/autotrain-animals-vs-humans2-37846100283")
8
+
9
+
10
 
11
  def predict(image):
12
  predictions = pipeline(image)