Update app.py

app.py

@@ -1,3 +1,16 @@
-import gradio as gr
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+import gradio
 
-gr.Interface.load('huggingface/vincentclaes/mit-indoor-scenes').launch(inbrowser=True)
+extractor = AutoFeatureExtractor.from_pretrained("vincentclaes/mit-indoor-scenes")
+model = AutoModelForImageClassification.from_pretrained("vincentclaes/mit-indoor-scenes")
+
+def classify(image):
+  inputs = extractor(images=image, return_tensors='pt')
+  outputs = model(**inputs)
+  last_hidden_states = outputs.last_hidden_state
+  return last_hidden_states
+
+gradio.Interface(fn=classify,
+                 inputs=gradio.inputs.Image(self, shape=(224,224), image_mode='RGB', source='upload', tool='editor', type='PIL.image', label=None, optional=False),
+                 outputs=gradio.outputs.Label(self, num_top_classes=5),
+                 allow_flagging='never').launch(inbrowser=True)