Update app.py

app.py

@@ -1,9 +1,15 @@
-from transformers import pipeline
-import gradio as gr
+from transformers import AutoImageProcessor, AutoBackbone
+import torch
+from PIL import Image
+import requests
+url = "http://images.cocodataset.org/val2017/000000039769.jpg"
+image = Image.open(requests.get(url, stream=True).raw)
+processor = AutoImageProcessor.from_pretrained("microsoft/swin-tiny-patch4-window7-224")
+model = AutoBackbone.from_pretrained("microsoft/swin-tiny-patch4-window7-224", out_indices=(1,))
 
-pipe = pipeline("image-classification", model="google/vit-base-patch16-224")
-
-gr.Interface.from_pipeline(pipe).launch()
+inputs = processor(image, return_tensors="pt")
+outputs = model(**inputs)
+feature_maps = outputs.feature_maps
 
 # import streamlit as st
 # from transformers import pipeline