LN1996 commited on
Commit
cb7c43d
1 Parent(s): 937ea93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import modal
2
 
3
- f = modal.Cls.lookup("casa-interior-hf-v1", "DesignModel")
4
 
5
  def casa_ai_run(dict=None, text=None):
6
 
@@ -11,7 +11,7 @@ def casa_ai_run(dict=None, text=None):
11
  print('Please provide context in form of image, audio, text')
12
  return None
13
 
14
- result_image = f.inference.remote(image, text)
15
  return result_image
16
 
17
  import gradio as gr
 
1
  import modal
2
 
3
+ f = modal.Cls.lookup("casa-interior-hf-v2", "DesignModel")
4
 
5
  def casa_ai_run(dict=None, text=None):
6
 
 
11
  print('Please provide context in form of image, audio, text')
12
  return None
13
 
14
+ result_image = f.inference.remote(image, text, mask)
15
  return result_image
16
 
17
  import gradio as gr