full downloading and launch

README.md

@@ -5,7 +5,7 @@ colorFrom: yellow
 colorTo: purple
 sdk: gradio
 sdk_version: 3.20.1
-app_file: server.py
+app_file: run.py
 pinned: false
 license: mit
 ---

run.py

@@ -0,0 +1,4 @@
+import os
+os.system('python download-model.py PygmalionAI/pygmalion-350m --branch main')
+# os.system('python download-model.py waifu-workshop/pygmalion-6b --branch original-sharded')
+os.system('python server.py --cpu --cai-chat --model pygmalion-350m')

server.py

@@ -5,9 +5,6 @@ import re
 import sys
 import time
 import zipfile
-import os
-os.system('cd text-generation-webui && python download-model.py PygmalionAI/pygmalion-350m --branch main')
-# os.system('cd text-generation-webui && python download-model.py waifu-workshop/pygmalion-6b --branch original-sharded')
 from pathlib import Path
 
 import gradio as gr