Spaces:

InklingSutra
/

flowers

Sleeping

InklingSutra commited on Jun 5, 2023

Commit

06783d8

1 Parent(s): 3b594e5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 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):

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