It works on my michane problem

app.py

@@ -6,6 +6,7 @@ from PIL import Image
 from torchvision import transforms
 import gradio as gr
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+print(device.type)
 class UPSC(nn.Module):
     def __init__(self):
         super(UPSC,self).__init__()
@@ -23,7 +24,7 @@ class UPSC(nn.Module):
         return self.model(x)
 
 model = UPSC().to(device)
-model.load_state_dict(torch.load("upscaling.pth", weights_only=True))
+model.load_state_dict(torch.load("upscaling.pth", weights_only=True,map_location=device))
 model.eval()
 
 def task(img):