Spaces:

ambarboza
/

flores

Sleeping

ambarboza commited on Jan 26

Commit

7a38213

•

1 Parent(s): 0ccd165

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import torch
 from transformers import AutoModelForImageClassification, AutoFeatureExtractor
 import gradio as gr
-model_id = f'ambarboza/vit-base-patch16-224-finetuned-flower'
 labels = ['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips']
 def classify_image(image):

 from transformers import AutoModelForImageClassification, AutoFeatureExtractor
 import gradio as gr
+model_id = f'jonathanfernandes/vit-base-patch16-224-finetuned-flower'
 labels = ['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips']
 def classify_image(image):