abdullahfurquan commited on
Commit
4949873
·
verified ·
1 Parent(s): 1df250f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -25,14 +25,14 @@ def generate_image_tool(prompt: str) -> str:
25
  prompt: A description of the image to generate.
26
 
27
  Returns:
28
- The generated image url.
29
 
30
  """
31
  try:
32
  image = image_generation_tool(prompt) # Get PIL Image
33
- file_path = "generated_image.jpg"
34
- image.save(file_path) # Save it
35
- return file_path # Return the file path
36
  except Exception as e:
37
  return f"Error generating image: {str(e)}"
38
 
 
25
  prompt: A description of the image to generate.
26
 
27
  Returns:
28
+ The generated image.
29
 
30
  """
31
  try:
32
  image = image_generation_tool(prompt) # Get PIL Image
33
+ # file_path = "generated_image.jpg"
34
+ # image.save(file_path) # Save it
35
+ return image.show()
36
  except Exception as e:
37
  return f"Error generating image: {str(e)}"
38