KeivanR commited on
Commit
c18aa66
·
verified ·
1 Parent(s): b0cd906

Update qwen_classifier/config.py

Browse files
Files changed (1) hide show
  1. qwen_classifier/config.py +3 -0
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
  ]