Spaces:
Sleeping
Sleeping
Update qwen_classifier/config.py
Browse files
qwen_classifier/config.py
CHANGED
@@ -4,6 +4,8 @@ import torch
|
|
4 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
5 |
|
6 |
# HF API config
|
|
|
|
|
7 |
TAG_NAMES = [
|
8 |
'games',
|
9 |
'geometry',
|
@@ -14,4 +16,5 @@ TAG_NAMES = [
|
|
14 |
'probabilities',
|
15 |
'strings',
|
16 |
'trees'
|
|
|
17 |
]
|
|
|
4 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
5 |
|
6 |
# HF API config
|
7 |
+
SPACE_URL = "https://keivanr-qwen-classifier-demo.hf.space"
|
8 |
+
|
9 |
TAG_NAMES = [
|
10 |
'games',
|
11 |
'geometry',
|
|
|
16 |
'probabilities',
|
17 |
'strings',
|
18 |
'trees'
|
19 |
+
|
20 |
]
|