Qdssa commited on
Commit
cf89ae1
·
verified ·
1 Parent(s): 5269db5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -18,10 +18,9 @@ from PIL import Image
18
  import io
19
 
20
  URL="http://127.0.0.1"
21
- OUTPUT_DIR = config('OUTPUT_DIR')
22
- INPUT_DIR = config('INPUT_DIR')
23
- COMF_PATH = config('COMF_PATH')
24
-
25
  import torch
26
 
27
  import spaces
@@ -85,7 +84,7 @@ def check_server_ready(port):
85
  def generate_image(image):
86
  prefix_filename = str(random.randint(0, 999999))
87
  prompt = read_prompt_from_file('good_upscaler.json')
88
- prompt = json.dumps(prompt, ensure_ascii=False).replace('ComfyUI', prefix_filename)
89
  prompt = json.loads(prompt)
90
 
91
  image = Image.fromarray(image)
@@ -95,7 +94,7 @@ def generate_image(image):
95
  new_port = str(random.randint(8123, 8200))
96
 
97
  try:
98
- process = subprocess.Popen([sys.executable, COMF_PATH, "--listen", "127.0.0.1", "--port", new_port])
99
  logger.debug(f'Subprocess started with PID: {process.pid}')
100
 
101
  for _ in range(40):
 
18
  import io
19
 
20
  URL="http://127.0.0.1"
21
+ OUTPUT_DIR="Backend/output"
22
+ INPUT_DIR="Backend/input"
23
+ BACKEND_PATH="Backend/main.py"
 
24
  import torch
25
 
26
  import spaces
 
84
  def generate_image(image):
85
  prefix_filename = str(random.randint(0, 999999))
86
  prompt = read_prompt_from_file('good_upscaler.json')
87
+ prompt = json.dumps(prompt, ensure_ascii=False).replace('OutPutImage', prefix_filename)
88
  prompt = json.loads(prompt)
89
 
90
  image = Image.fromarray(image)
 
94
  new_port = str(random.randint(8123, 8200))
95
 
96
  try:
97
+ process = subprocess.Popen([sys.executable, BACKEND_PATH, "--listen", "127.0.0.1", "--port", new_port])
98
  logger.debug(f'Subprocess started with PID: {process.pid}')
99
 
100
  for _ in range(40):