Caniya commited on
Commit
9089903
1 Parent(s): e7f11d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import Pytorch
2
  from transformers import AutoModelForImageClassification, AutoFeatureExtractor
3
  import gradio as gr
4
 
@@ -17,5 +17,5 @@ def classify_image(image):
17
 
18
  interface = gr.Interface(fn=classify_image,
19
  inputs='image',
20
- examples=['dandelion_1.jpg', 'Dandelion_2.jpg'],
21
  outputs='label').launch()
 
1
+ import torch
2
  from transformers import AutoModelForImageClassification, AutoFeatureExtractor
3
  import gradio as gr
4
 
 
17
 
18
  interface = gr.Interface(fn=classify_image,
19
  inputs='image',
20
+ examples=['Dandelion_2.jpg', 'dandelion_1.jpg'],
21
  outputs='label').launch()