xpvwklvjp commited on
Commit
6d60b67
·
verified ·
1 Parent(s): 7fdd80f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -30,6 +30,7 @@ def generate_image(prompt, max_retries=5):
30
  result = demo(prompt)
31
  generation_time = time.time() - start_time
32
  logger.info(f"Generation completed in {generation_time:.2f} seconds")
 
33
  return result, f"Image generated successfully\nGeneration time: {generation_time:.2f} seconds"
34
  except Exception as e:
35
  logger.error(f"Error on attempt {attempt + 1}: {str(e)}")
 
30
  result = demo(prompt)
31
  generation_time = time.time() - start_time
32
  logger.info(f"Generation completed in {generation_time:.2f} seconds")
33
+ logger.info(result)
34
  return result, f"Image generated successfully\nGeneration time: {generation_time:.2f} seconds"
35
  except Exception as e:
36
  logger.error(f"Error on attempt {attempt + 1}: {str(e)}")