Spaces:
Running
Running
Commit
·
134aae6
1
Parent(s):
c59ebda
fixed aragpt2 links
Browse files- backend/services.py +4 -4
backend/services.py
CHANGED
@@ -24,10 +24,10 @@ class TextGeneration:
|
|
24 |
# "aragpt2-base": "D:/ML/Models/aragpt2-base",
|
25 |
# }
|
26 |
self.model_names_or_paths = {
|
27 |
-
"aragpt2-medium": "
|
28 |
-
"aragpt2-base": "
|
29 |
-
"aragpt2-large": "
|
30 |
-
"aragpt2-mega": "
|
31 |
}
|
32 |
set_seed(42)
|
33 |
|
|
|
24 |
# "aragpt2-base": "D:/ML/Models/aragpt2-base",
|
25 |
# }
|
26 |
self.model_names_or_paths = {
|
27 |
+
"aragpt2-medium": "aubmindlab/aragpt2-medium",
|
28 |
+
"aragpt2-base": "aubmindlab/aragpt2-base",
|
29 |
+
"aragpt2-large": "aubmindlab/aragpt2-large",
|
30 |
+
"aragpt2-mega": "aubmindlab/aragpt2-mega",
|
31 |
}
|
32 |
set_seed(42)
|
33 |
|