RishubhPar commited on
Commit
977312d
·
verified ·
1 Parent(s): 63a2e29
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -128,8 +128,6 @@ def load_pipeline_single_gpu():
128
 
129
 
130
  # Initializing the pipeline with gpu
131
- print("INIT pipeline with the gpu")
132
- status = load_pipeline_single_gpu()
133
  print("[main] worker init status:", status)
134
 
135
  # -----------------------------
@@ -333,6 +331,8 @@ def generate_image_stack_edits(text_prompt, n_edits, input_image):
333
  global PIPELINE
334
  if PIPELINE is None:
335
  status = load_pipeline_single_gpu()
 
 
336
 
337
  if not input_image or not text_prompt or text_prompt.startswith("Please select"):
338
  return [], None
 
128
 
129
 
130
  # Initializing the pipeline with gpu
 
 
131
  print("[main] worker init status:", status)
132
 
133
  # -----------------------------
 
331
  global PIPELINE
332
  if PIPELINE is None:
333
  status = load_pipeline_single_gpu()
334
+
335
+ print("loaded pipeline status: {}".format(status))
336
 
337
  if not input_image or not text_prompt or text_prompt.startswith("Please select"):
338
  return [], None