wissamantoun commited on
Commit
134aae6
1 Parent(s): c59ebda

fixed aragpt2 links

Browse files
Files changed (1) hide show
  1. 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": "https://huggingface.co/aubmindlab/aragpt2-medium",
28
- "aragpt2-base": "https://huggingface.co/aubmindlab/aragpt2-base",
29
- "aragpt2-large": "https://huggingface.co/aubmindlab/aragpt2-large",
30
- "aragpt2-mega": "https://huggingface.co/aubmindlab/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