Spaces:
Runtime error
Runtime error
Commit ·
da852e5
1
Parent(s): 8ee03b1
Update app/main.py
Browse files- app/main.py +1 -1
app/main.py
CHANGED
|
@@ -10,7 +10,7 @@ except Exception as e:
|
|
| 10 |
print(e)
|
| 11 |
|
| 12 |
# Config
|
| 13 |
-
MODELS_PATH = "kadabengaran/
|
| 14 |
|
| 15 |
id2label= {0: 'Other', 1: 'Problem Discovery', 2: 'Information Seeking', 3: 'Feature Request'}
|
| 16 |
label2id= {'Other': 0, 'Problem Discovery': 1, 'Information Seeking': 2, 'Feature Request': 3}
|
|
|
|
| 10 |
print(e)
|
| 11 |
|
| 12 |
# Config
|
| 13 |
+
MODELS_PATH = "kadabengaran/distilbert-base-uncased-lora-text-classification"
|
| 14 |
|
| 15 |
id2label= {0: 'Other', 1: 'Problem Discovery', 2: 'Information Seeking', 3: 'Feature Request'}
|
| 16 |
label2id= {'Other': 0, 'Problem Discovery': 1, 'Information Seeking': 2, 'Feature Request': 3}
|