Spaces:
Build error
Build error
Quyet
commited on
Commit
·
617fa8c
1
Parent(s):
c0399e9
fix gradio bug
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import re, time
|
| 2 |
import matplotlib.pyplot as plt
|
| 3 |
from threading import Timer
|
|
@@ -197,6 +198,10 @@ def chat(message, history):
|
|
| 197 |
if __name__ == '__main__':
|
| 198 |
# euc_100()
|
| 199 |
# euc_200('I am happy about my academic record.')
|
|
|
|
|
|
|
|
|
|
|
|
|
| 200 |
title = "PsyPlus Empathetic Chatbot"
|
| 201 |
description = "Gradio demo for product of PsyPlus. Based on rule-based CBT and conversational AI model DialoGPT"
|
| 202 |
iface = gr.Interface(
|
|
@@ -208,4 +213,7 @@ if __name__ == '__main__':
|
|
| 208 |
title=title,
|
| 209 |
description=description,
|
| 210 |
)
|
| 211 |
-
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import argparse
|
| 2 |
import re, time
|
| 3 |
import matplotlib.pyplot as plt
|
| 4 |
from threading import Timer
|
|
|
|
| 198 |
if __name__ == '__main__':
|
| 199 |
# euc_100()
|
| 200 |
# euc_200('I am happy about my academic record.')
|
| 201 |
+
parser = argparse.ArgumentParser()
|
| 202 |
+
parser.add_argument('--run_share_mode', type=int, default=0, help='if test on own server, need to use share mode')
|
| 203 |
+
args = parser.parse_args()
|
| 204 |
+
|
| 205 |
title = "PsyPlus Empathetic Chatbot"
|
| 206 |
description = "Gradio demo for product of PsyPlus. Based on rule-based CBT and conversational AI model DialoGPT"
|
| 207 |
iface = gr.Interface(
|
|
|
|
| 213 |
title=title,
|
| 214 |
description=description,
|
| 215 |
)
|
| 216 |
+
if args.run_share_mode == 0:
|
| 217 |
+
iface.launch(debug=True)
|
| 218 |
+
else:
|
| 219 |
+
iface.launch(debug=True, server_name="0.0.0.0", server_port=2022, share=True)
|