Hugging Face's logo

Spaces:
aubmindlab
/
Arabic-NLP
Running

App Files Community
1
wissamantoun commited on
Commit
5d70a9c
·
1 Parent(s): 27a1a2f

fixes

Browse files
Files changed (1) hide show
  1. backend/services.py +2 -2
backend/services.py CHANGED
@@ -45,8 +45,8 @@ class TextGeneration:
45
  # "aragpt2-base": "D:/ML/Models/aragpt2-base",
46
  # }
47
  self.model_names_or_paths = {
48
- "aragpt2-medium": "aubmindlab/aragpt2-medium",
49
- # "aragpt2-base": "aubmindlab/aragpt2-base",
50
  # "aragpt2-large": "aubmindlab/aragpt2-large",
51
  "aragpt2-mega": "aubmindlab/aragpt2-mega",
52
  }
 
45
  # "aragpt2-base": "D:/ML/Models/aragpt2-base",
46
  # }
47
  self.model_names_or_paths = {
48
+ # "aragpt2-medium": "aubmindlab/aragpt2-medium",
49
+ "aragpt2-base": "aubmindlab/aragpt2-base",
50
  # "aragpt2-large": "aubmindlab/aragpt2-large",
51
  "aragpt2-mega": "aubmindlab/aragpt2-mega",
52
  }