AMfeta99 commited on
Commit
e800bcd
·
verified ·
1 Parent(s): 8d472c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -105,7 +105,13 @@ def generate_object_history(object_name):
105
  try:
106
  result = agent.run(
107
  general_instruction,
108
- additional_args={"user_prompt": prompt}
 
 
 
 
 
 
109
  )
110
 
111
  # result is tuple: (filepath, seed)
 
105
  try:
106
  result = agent.run(
107
  general_instruction,
108
+ additional_args={"prompt": prompt,
109
+ "width": 256, # specify width
110
+ "height": 256, # specify height
111
+ "seed": 0, # optional seed
112
+ "randomize_seed": False, # optional
113
+ "num_inference_steps": 4 # optional
114
+ }
115
  )
116
 
117
  # result is tuple: (filepath, seed)