fix to device

app.py

@@ -12,7 +12,8 @@ import warnings
 warnings.filterwarnings("ignore")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model, preprocess = clip.load('ViT-L/14@336px').to(device)
+model, preprocess = clip.load('ViT-L/14@336px')
+model.to(device)
 
 
 def zeroshot_detection(Press_Clear_Dont_Stack_Image):
@@ -35,3 +36,4 @@ def zeroshot_detection(Press_Clear_Dont_Stack_Image):
 gr.Interface(fn=zeroshot_detection, 
              inputs=[gr.Image(type="pil")],
              outputs=gr.Label(num_top_classes=1)).launch()
+