SahilCarterr commited on
Commit
6cedc33
·
verified ·
1 Parent(s): 62af58d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -41,15 +41,13 @@ def load_models():
41
 
42
  return model
43
 
44
- @gr.CachedFunction()
45
- def load_model_cached():
46
- return load_models()
47
 
 
48
  @spaces.GPU
49
  def process_and_generate(editor_component, reference_image, prompt, height, width,
50
  scale, steps=50, seed=42, use_context=True, num_images=1):
51
  print('Begin processing!')
52
- model = load_model_cached()
53
  # Job directory.
54
  job_name = datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
55
  job_dir = os.path.join(SAVE_DIR, job_name)
 
41
 
42
  return model
43
 
 
 
 
44
 
45
+ model = load_models()
46
  @spaces.GPU
47
  def process_and_generate(editor_component, reference_image, prompt, height, width,
48
  scale, steps=50, seed=42, use_context=True, num_images=1):
49
  print('Begin processing!')
50
+
51
  # Job directory.
52
  job_name = datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
53
  job_dir = os.path.join(SAVE_DIR, job_name)