fixed sketchpad

app.py

@@ -2,6 +2,7 @@ from pathlib import Path
 import torch
 import gradio as gr
 from torch import nn
+import cv2
 
 LABELS = Path("class_names.txt").read_text().splitlines()
 
@@ -20,6 +21,7 @@ model = nn.Sequential(
     nn.ReLU(),
     nn.Linear(256, len(LABELS)),
 )
+ 
 state_dict = torch.load("pytorch_model.bin", map_location="cpu")
 model.load_state_dict(state_dict, strict=False)
 model.eval()
@@ -27,17 +29,20 @@ model.eval()
 
 def predict(im):    
     if(im is not None):
-        x = torch.tensor(im, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.0
+        # print(im.size)
+        im = cv2.resize(im, dsize=(28, 28), interpolation=cv2.INTER_CUBIC)
+        x = torch.tensor(im, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.0 
+        # print(type(im), im.size)
         with torch.no_grad():
             out = model(x)
         probabilities = torch.nn.functional.softmax(out[0], dim=0)
         values, indices = torch.topk(probabilities, 5)
         return {LABELS[i]: v.item() for i, v in zip(indices, values)}
 
-
+sp = gr.Sketchpad(shape=(56,56),brush_radius =1)
 interface = gr.Interface(
 predict,
-inputs="sketchpad",
+inputs=sp,
 outputs="label",
 theme="huggingface",
 title="Reconocimiento de garabtosss",
@@ -45,4 +50,4 @@ description="Querés jugar al Picdictonary con una AI? Dibujá un objeto cotidia
 article="<p style='text-align: center'>Reconocimiento de Garabatos | Demo Model</p>",
 live=True,
 )
-interface.launch()#share=True
+interface.launch()

hugg

@@ -0,0 +1,7 @@
+-----BEGIN OPENSSH PRIVATE KEY-----
+b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW
+QyNTUxOQAAACDWyBwj820SKGSnQe8shCizpUiPbMaXH4H9PMdRjK4zlgAAAKBssbVYbLG1
+WAAAAAtzc2gtZWQyNTUxOQAAACDWyBwj820SKGSnQe8shCizpUiPbMaXH4H9PMdRjK4zlg
+AAAEAUxfh4KZ+ntU7ttAHhYLz9Va3gl7UoTLLvYVo/+Mo49tbIHCPzbRIoZKdB7yyEKLOl
+SI9sxpcfgf08x1GMrjOWAAAAFm1yb2xhbmRvQGNlaWJhbC5lZHUudXkBAgMEBQYH
+-----END OPENSSH PRIVATE KEY-----

hugg.pub

@@ -0,0 +1 @@
+ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAINbIHCPzbRIoZKdB7yyEKLOlSI9sxpcfgf08x1GMrjOW mrolando@ceibal.edu.uy