Hugging Face's logo

Spaces:
rsadaphule
/
flowers
Sleeping

App Files Community
rsadaphule commited on
Commit
201463a
•
1 Parent(s): 57c8a52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  from transformers import AutoModelForImageClassification, AutoFeatureExtractor
3
  import gradio as gr
4
 
5
- model_id = f'vit-base-patch16-224-finetuned-flower-ravindrasadaphule'
6
  labels = ['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips']
7
 
8
  def classify_image(image):
 
2
  from transformers import AutoModelForImageClassification, AutoFeatureExtractor
3
  import gradio as gr
4
 
5
+ model_id = f'rsadaphule/vit-base-patch16-224-finetuned-flower-ravindrasadaphule'
6
  labels = ['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips']
7
 
8
  def classify_image(image):