radinhas commited on
Commit
02e2c20
1 Parent(s): 1980208

Update apis/chat_api.py

Browse files
Files changed (1) hide show
  1. apis/chat_api.py +4 -4
apis/chat_api.py CHANGED
@@ -45,11 +45,11 @@ class ChatAPIApp:
45
 
46
  class TranslateCompletionsPostItem(BaseModel):
47
  from_language: str = Field(
48
- default="auto",
49
  description="(str) `Detect`",
50
  )
51
  to_language: str = Field(
52
- default="en",
53
  description="(str) `en`",
54
  )
55
  input_text: str = Field(
@@ -107,7 +107,7 @@ class ChatAPIApp:
107
  time_start = time.time()
108
  #TRANSFORMERS_CACHE
109
  pretrained_model = "facebook/m2m100_1.2B"
110
- cache_dir = "models/"
111
  tokenizer = M2M100Tokenizer.from_pretrained(pretrained_model, cache_dir=cache_dir)
112
  model = M2M100ForConditionalGeneration.from_pretrained(
113
  pretrained_model, cache_dir=cache_dir
@@ -139,7 +139,7 @@ class ChatAPIApp:
139
 
140
  class DetectLanguagePostItem(BaseModel):
141
  input_text: str = Field(
142
- default="Hello",
143
  description="(str) `Text for detection`",
144
  )
145
 
 
45
 
46
  class TranslateCompletionsPostItem(BaseModel):
47
  from_language: str = Field(
48
+ default="en",
49
  description="(str) `Detect`",
50
  )
51
  to_language: str = Field(
52
+ default="fa",
53
  description="(str) `en`",
54
  )
55
  input_text: str = Field(
 
107
  time_start = time.time()
108
  #TRANSFORMERS_CACHE
109
  pretrained_model = "facebook/m2m100_1.2B"
110
+ cache_dir = "apis/models/"
111
  tokenizer = M2M100Tokenizer.from_pretrained(pretrained_model, cache_dir=cache_dir)
112
  model = M2M100ForConditionalGeneration.from_pretrained(
113
  pretrained_model, cache_dir=cache_dir
 
139
 
140
  class DetectLanguagePostItem(BaseModel):
141
  input_text: str = Field(
142
+ default="Hello, how are you?",
143
  description="(str) `Text for detection`",
144
  )
145