danieldux commited on
Commit
6b9f417
·
verified ·
1 Parent(s): 89acf17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ huggingface_hub.login(token=token)
10
  # gr.load("models/ICILS/xlm-r-icils-ilo", hf_token=token).launch()
11
 
12
  # Load the pre-trained model
13
- classifier = pipeline("text-classification", model="ICILS/xlm-r-icils-ilo", use_auth_token=token, device='cuda:0')
14
 
15
  # Define the prediction function
16
  @spaces.GPU
 
10
  # gr.load("models/ICILS/xlm-r-icils-ilo", hf_token=token).launch()
11
 
12
  # Load the pre-trained model
13
+ classifier = pipeline("text-classification", model="ICILS/xlm-r-icils-ilo", use_auth_token=True, device='cuda:0')
14
 
15
  # Define the prediction function
16
  @spaces.GPU