Updated space

app.py

@@ -2,13 +2,22 @@ import gradio as gr
 import spaces
 import torch
 
+from PIL import Image
+import requests
+from transformers import DetrImageProcessor
+
+
 zero = torch.Tensor([0]).cuda()
 print(zero.device) # <-- 'cpu' 🤔
 
 @spaces.GPU
-def greet(n):
+def greet(image):
+    processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
+
+    encoding = processor(image, return_tensors='pt')
+    print(encoding.keys())
     print(zero.device) # <-- 'cuda:0' 🤗
     return f"Hello {zero + n} Tensor"
 
-demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
+demo = gr.Interface(fn=greet, inputs=gr.Image(label="Input image", type="pil"), outputs=gr.Image(label="Output image", type="pil"))
 demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+transformers 
+timm
+torch