Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ token = os.getenv("HF_TOKEN")
|
|
| 11 |
# gr.load("models/ICILS/xlm-r-icils-ilo", hf_token=token).launch()
|
| 12 |
|
| 13 |
# Load the pre-trained model
|
| 14 |
-
classifier = pipeline("text-classification", model="ICILS/xlm-r-icils-ilo",
|
| 15 |
|
| 16 |
# Define the prediction function
|
| 17 |
@spaces.GPU
|
|
|
|
| 11 |
# gr.load("models/ICILS/xlm-r-icils-ilo", hf_token=token).launch()
|
| 12 |
|
| 13 |
# Load the pre-trained model
|
| 14 |
+
classifier = pipeline("text-classification", model="ICILS/xlm-r-icils-ilo", use_auth_token=token)
|
| 15 |
|
| 16 |
# Define the prediction function
|
| 17 |
@spaces.GPU
|