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 |
|