[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,
|