Update app.py

app.py

@@ -17,8 +17,6 @@ model = AutoModelForImageClassification.from_pretrained('0-ma/vit-geometric-shap
 
 
 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")
     logits = model(**inputs)['logits'].cpu().detach().numpy()[0]
     logits_positive = logits
@@ -31,10 +29,8 @@ def predict(image):
     return confidences
 
  
-    #return {"predicted_label" : predicted_labels[0] }
- 
 title = "Geometric Shape Classifier"
-description = "A geometric shape setector."
+description = "The geometric shape classifier: 0-ma/vit-geometric-shapes-tiny."
 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']