Update app.py

app.py

@@ -17,9 +17,7 @@ feature_extractor = AutoImageProcessor.from_pretrained('0-ma/vit-geometric-shape
 model = AutoModelForImageClassification.from_pretrained('0-ma/vit-geometric-shapes-tiny')
 
 labels = []
-def predict(img):
-    image = PILImage.create(img)
-
+def predict(image):
     feature_extractor = AutoImageProcessor.from_pretrained('0-ma/vit-geometric-shapes-tiny')
     model = AutoModelForImageClassification.from_pretrained('0-ma/vit-geometric-shapes-tiny')
     inputs = feature_extractor(images=[image], return_tensors="pt")
@@ -36,4 +34,4 @@ description = "A geometric shape setector."
 examples = ['example/1_None.jpg','example/2_Circle.jpg','example/3_Triangle.jpg','example/4_Square.jpg','example/5_Pentagone.jpg','example/6_Hexagone.jpg']
 
 
-gr.Interface(fn=predict,inputs=gr.Image(),outputs=gr.Label(),title=title,description=description,examples=examples).launch()
+gr.Interface(fn=predict,inputs=gr.Image(type="pil"),outputs=gr.Label(),title=title,description=description,examples=examples).launch()