hamdan07 commited on
Commit
f260a52
1 Parent(s): bb478f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -18,11 +18,11 @@ head = (
18
  )
19
 
20
  def predict_input_image(img):
21
- from transformers import AutoFeatureExtractor, AutoModelForImageClassification
22
 
23
- extractor = AutoFeatureExtractor.from_pretrained("swww/test")
24
 
25
- model = AutoModelForImageClassification.from_pretrained("swww/test") }
26
 
27
 
28
  image = gr.inputs.Image(shape=(500, 500), image_mode='L', invert_colors=False, source="upload")
 
18
  )
19
 
20
  def predict_input_image(img):
21
+ from transformers import AutoFeatureExtractor, AutoModelForImageClassification
22
 
23
+ extractor = AutoFeatureExtractor.from_pretrained("swww/test")
24
 
25
+ model = AutoModelForImageClassification.from_pretrained("swww/test") }
26
 
27
 
28
  image = gr.inputs.Image(shape=(500, 500), image_mode='L', invert_colors=False, source="upload")