LN1996 commited on
Commit
0282d2d
·
verified ·
1 Parent(s): 9866f53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -6,7 +6,6 @@ import requests
6
  import modal
7
  import PIL
8
 
9
- f = modal.Cls.lookup("casa-interior-hf-v4", "DesignModel")
10
  f_gc = modal.Cls.lookup("casa-interior-gc-v2", "GetProduct")
11
 
12
  def casa_ai_run_tab1(image=None, text=None):
@@ -19,6 +18,7 @@ def casa_ai_run_tab1(image=None, text=None):
19
  print('Please provide a text prompt')
20
  return None
21
 
 
22
  result_image = f.inference.remote("tab1", image, text)
23
  return result_image
24
 
@@ -36,8 +36,9 @@ def casa_ai_run_tab2(dict=None, text=None):
36
  if image is None and text is None:
37
  print('Please provide context in form of image, text')
38
  return None
39
-
40
- result_image = f.inference.remote("tab2", image, text, mask)
 
41
  return result_image
42
 
43
  def casa_ai_run_tab3(dict=None):
 
6
  import modal
7
  import PIL
8
 
 
9
  f_gc = modal.Cls.lookup("casa-interior-gc-v2", "GetProduct")
10
 
11
  def casa_ai_run_tab1(image=None, text=None):
 
18
  print('Please provide a text prompt')
19
  return None
20
 
21
+ f = modal.Cls.lookup("casa-interior-hf-v4", "DesignModel")
22
  result_image = f.inference.remote("tab1", image, text)
23
  return result_image
24
 
 
36
  if image is None and text is None:
37
  print('Please provide context in form of image, text')
38
  return None
39
+
40
+ f_tab2 = modal.Cls.lookup("casa-interior-hf-v3", "DesignModel")
41
+ result_image = f_tab2.inference.remote("tab2", image, text, mask)
42
  return result_image
43
 
44
  def casa_ai_run_tab3(dict=None):