antonovmaxim commited on
Commit
973eea2
1 Parent(s): f5f216c

full downloading and launch

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. run.py +4 -0
  3. server.py +0 -3
README.md CHANGED
@@ -5,7 +5,7 @@ colorFrom: yellow
5
  colorTo: purple
6
  sdk: gradio
7
  sdk_version: 3.20.1
8
- app_file: server.py
9
  pinned: false
10
  license: mit
11
  ---
 
5
  colorTo: purple
6
  sdk: gradio
7
  sdk_version: 3.20.1
8
+ app_file: run.py
9
  pinned: false
10
  license: mit
11
  ---
run.py ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ import os
2
+ os.system('python download-model.py PygmalionAI/pygmalion-350m --branch main')
3
+ # os.system('python download-model.py waifu-workshop/pygmalion-6b --branch original-sharded')
4
+ os.system('python server.py --cpu --cai-chat --model pygmalion-350m')
server.py CHANGED
@@ -5,9 +5,6 @@ import re
5
  import sys
6
  import time
7
  import zipfile
8
- import os
9
- os.system('cd text-generation-webui && python download-model.py PygmalionAI/pygmalion-350m --branch main')
10
- # os.system('cd text-generation-webui && python download-model.py waifu-workshop/pygmalion-6b --branch original-sharded')
11
  from pathlib import Path
12
 
13
  import gradio as gr
 
5
  import sys
6
  import time
7
  import zipfile
 
 
 
8
  from pathlib import Path
9
 
10
  import gradio as gr