Add application file

app.py

@@ -0,0 +1,30 @@
+from transformers import AutoImageProcessor, AutoModelForImageClassification
+import gradio as gr
+
+def classifyArchitecture(img):
+    # Call the model
+    image_processor = AutoImageProcessor.from_pretrained("hanslab37/architectural_styles_classifier")
+    classifier = AutoModelForImageClassification.from_pretrained("hanslab37/architectural_styles_classifier")
+
+    # Transform our image and pass it through the model
+    inputs = image_processor(image, return_tensors="pt")
+    output = classifier(**inputs)
+
+    # Display the image
+    display(img)
+
+    # Predicted Class probabilities
+    proba = output.logits.softmax(1)
+    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.launch()