Update app.py

app.py

@@ -2,13 +2,15 @@ from fastai.vision.all import *
 import gradio as gr
 import cloudpickle
 import numpy as np
-device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
+import torch
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 
 # Cargamos el learner
 def load(map_location='cpu', pickle_module=pickle, **pickle_load_args):
     with open ('modelomemes.pkl',mode='rb') as file:
-        learn=torch.load(cloudpickle.load(file), map_location="cpu")
+        learn=cloudpickle.load(file).to(device)
     
 load()