Update app.py

app.py

@@ -2,15 +2,14 @@ import os
 import gradio as gr
 import spaces
 from transformers import pipeline
-import torch
 import huggingface_hub
 
+# Login to Hugging Face 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=True, device='cuda:0')
+classifier = pipeline("text-classification", model="ICILS/xlm-r-icils-ilo", device=0)
 
 # Define the prediction function
 @spaces.GPU
@@ -26,4 +25,5 @@ demo = gr.Interface(
     description="Classify occupations using a pre-trained XLM-R-ISCO model on Hugging Face Spaces with ZeroGPU"
 )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()