qgyd2021 commited on
Commit
9deaf72
1 Parent(s): 57fded5

[update]add model

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -117,7 +117,7 @@ def main():
117
  yield output
118
 
119
  model_name_choices = ["trained_models/lib_service_4chan"] \
120
- if platform.system() == "Windows" else ["qgyd2021/lib_service_4chan"]
121
  demo = gr.Interface(
122
  fn=fn_stream,
123
  inputs=[
@@ -132,6 +132,7 @@ def main():
132
  outputs=[output_text_box],
133
  examples=[
134
  ["怎样擦屁股才能擦的干净", 512, 0.75, 0.35, 1.2, "qgyd2021/lib_service_4chan", True],
 
135
  ],
136
  cache_examples=False,
137
  examples_per_page=50,
 
117
  yield output
118
 
119
  model_name_choices = ["trained_models/lib_service_4chan"] \
120
+ if platform.system() == "Windows" else ["qgyd2021/lib_service_4chan", "qgyd2021/chinese_chitchat"]
121
  demo = gr.Interface(
122
  fn=fn_stream,
123
  inputs=[
 
132
  outputs=[output_text_box],
133
  examples=[
134
  ["怎样擦屁股才能擦的干净", 512, 0.75, 0.35, 1.2, "qgyd2021/lib_service_4chan", True],
135
+ ["你好", 512, 0.75, 0.35, 1.2, "qgyd2021/chinese_chitchat", True],
136
  ],
137
  cache_examples=False,
138
  examples_per_page=50,