Update app.py

app.py

@@ -3,15 +3,14 @@ import gradio as gr
 import spaces
 from transformers import pipeline
 import torch
-
-zero = torch.Tensor([0]).cuda()
-print(zero.device) # <-- 'cpu' 🤔
+import huggingface_hub
 
 token = os.getenv("HF_TOKEN")
+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)
+classifier = pipeline("text-classification", model="ICILS/xlm-r-icils-ilo", use_auth_token=token, device=torch.device)
 
 # Define the prediction function
 @spaces.GPU