Update app.py

app.py

@@ -33,7 +33,7 @@ def predict_and_show(image):
         outputs = vgg16(image_tensor)
         _, predicted = torch.max(outputs.data, 1)
 
-    class_names = ["Class1", "Class2", "Class3"]  # Replace with your actual class names
+    class_names = ["Classic", "Modern", "Vintage", "Glamour", "Scandinavian", "Rustic", "ArtDeco", "Industrial"] 
     predicted_label = class_names[predicted.item()]
 
     plt.imshow(image)