[update]add model
Browse files
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,
|