Rahatara commited on
Commit
3736dbd
·
verified ·
1 Parent(s): 0cc1d8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -79,8 +79,8 @@ def inpaint_defect(image, prompt, num_images=1):
79
 
80
  for _ in range(num_images):
81
  input = {
82
- "image": image_data_url,
83
- "prompt": prompt,
84
  "scheduler": "K_EULER_ANCESTRAL",
85
  "num_outputs": 1,
86
  "guidance_scale": 7.5,
@@ -90,7 +90,7 @@ def inpaint_defect(image, prompt, num_images=1):
90
 
91
  prediction = rep_client.predictions.create(
92
  version="10e63b0e6361eb23a0374f4d9ee145824d9d09f7a31dcd70803193ebc7121430",
93
- input = input
94
  )
95
  prediction.wait()
96
  if prediction.status == "succeeded":
@@ -203,4 +203,4 @@ with gr.Blocks() as app:
203
  )
204
 
205
  if __name__ == "__main__":
206
- app.launch()
 
79
 
80
  for _ in range(num_images):
81
  input = {
82
+ "input_image": image_data_url,
83
+ "instruction_text": prompt,
84
  "scheduler": "K_EULER_ANCESTRAL",
85
  "num_outputs": 1,
86
  "guidance_scale": 7.5,
 
90
 
91
  prediction = rep_client.predictions.create(
92
  version="10e63b0e6361eb23a0374f4d9ee145824d9d09f7a31dcd70803193ebc7121430",
93
+ input=input
94
  )
95
  prediction.wait()
96
  if prediction.status == "succeeded":
 
203
  )
204
 
205
  if __name__ == "__main__":
206
+ app.launch()