Update app.py

app.py

@@ -10,7 +10,7 @@ huggingface_hub.login(token=token)
 # gr.load("models/ICILS/xlm-r-icils-ilo", hf_token=token).launch()
 
 # Load the pre-trained model
-classifier = pipeline("text-classification", model="ICILS/xlm-r-icils-ilo", use_auth_token=token, device=torch.device)
+classifier = pipeline("text-classification", model="ICILS/xlm-r-icils-ilo", use_auth_token=token, device='cuda:0')
 
 # Define the prediction function
 @spaces.GPU