joramar commited on
Commit
2d52f0e
1 Parent(s): bbf54e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -53,6 +53,9 @@ def generate(prompt, uploaded_file, prompt_addition, negative_prompt, num_images
53
  img_preprocessed, error_message = preprocess_image(uploaded_file)
54
  if error_message:
55
  return error_message
 
 
 
56
  params = {
57
  "image": img_preprocessed,
58
  "num_inference_steps": num_steps,
 
53
  img_preprocessed, error_message = preprocess_image(uploaded_file)
54
  if error_message:
55
  return error_message
56
+
57
+ num_images = int(num_images)
58
+
59
  params = {
60
  "image": img_preprocessed,
61
  "num_inference_steps": num_steps,