older gradio, torch

README.md

@@ -4,7 +4,7 @@ emoji: 💯
 colorFrom: indigo
 colorTo: gray
 sdk: gradio
-sdk_version: 4.8.0
+sdk_version: 4.5.0
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -1,17 +1,17 @@
 import gradio as gr
 import numpy as np
+import torch
+import torchvision
 
 def on_submit(img):
-    img = img['composite']
-    print(img.shape)
-    print(img.dtype)
-    print(img[:3, :3])
-    return f"Sum: {np.sum(img)}"
+    img = img['composite'].astype(np.float32)
+    img = torch.from_numpy(img)
+    return f"Sum: {img.shape}"
 
 iface = gr.Interface(
     title = "LeNet",
     fn = on_submit,
-    inputs=gr.Sketchpad(width=280, height=280, image_mode='P'),
+    inputs=gr.Sketchpad(image_mode='P'),
     outputs=gr.Label(),
 )
 iface.launch()