Hugging Face's logo

Spaces:
ALVHB95
/
TFM_DataScience_APP
Running

App Files Community
ALVHB95 commited on
Commit
e87b593
·
1 Parent(s): dfc8ff7

app14

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ from pydantic import BaseModel
31
  import shutil
32
 
33
  # Cell 1: Image Classification Model
34
- image_pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
35
 
36
  def predict_image(input_img):
37
  predictions = image_pipeline(input_img)
 
31
  import shutil
32
 
33
  # Cell 1: Image Classification Model
34
+ image_pipeline = pipeline(task="image-classification", model="microsoft/resnet-50")
35
 
36
  def predict_image(input_img):
37
  predictions = image_pipeline(input_img)