Update app.py

app.py

@@ -13,8 +13,8 @@ labels =  [
 ] 
 #images = [Image.open(requests.get("https://raw.githubusercontent.com/0-ma/geometric-shape-detector/main/input/exemple_circle.jpg", stream=True).raw), 
 #        Image.open(requests.get("https://raw.githubusercontent.com/0-ma/geometric-shape-detector/main/input/exemple_pentagone.jpg", stream=True).raw)]
-feature_extractor = AutoImageProcessor.from_pretrained('0-ma/swin-geometric-shapes-tiny')
-model = AutoModelForImageClassification.from_pretrained('0-ma/swin-geometric-shapes-tiny')
+feature_extractor = AutoImageProcessor.from_pretrained('0-ma/vit-geometric-shapes-tiny')
+model = AutoModelForImageClassification.from_pretrained('0-ma/vit-geometric-shapes-tiny')
 
    
 print(predicted_labels)