Fix bug in gradio

app.py

@@ -7,7 +7,7 @@ def classifyArchitecture(img):
     classifier = AutoModelForImageClassification.from_pretrained("hanslab37/architectural_styles_classifier")
 
     # Transform our image and pass it through the model
-    inputs = image_processor(image, return_tensors="pt")
+    inputs = image_processor(img, return_tensors="pt")
     output = classifier(**inputs)
 
     # Display the image
@@ -18,13 +18,8 @@ def classifyArchitecture(img):
     preds = proba.argmax(1)
     
     result = f'Predicted Architectural Styles: {classifier.config.id2label[preds.item()]} (Confidence: {proba[0][preds.item()].item()})'
-    # result = '''{
-    #     score: {0},
-    #     label: {1}
-    # }
-    # '''.format(proba[0][preds.item()].item(), classifier.config.id2label[preds.item()])
-    # epred = "Label: {0}, Score: {1}".format(classifier.config.id2label[i], proba[0][i].item())
+
     return result
 
-page = gr.Interface(fn=classifyArchitecture, inputs="image", outputs="text")
+page = gr.Interface(fn=classifyArchitecture, inputs="image", outputs="text", title="Classify Architecture Styles",)
 page.launch()