Update app.py

app.py

@@ -32,9 +32,9 @@ def process(prompt):
     print(time)
     time = time.replace(".", "").replace(":","")
     image = client.text_to_image(response)
-    image.save("outputs/images/" + time)
+    image.save("outputs/images/" + time + ".png")
     with open("outputs/" + time + ".json", "wb") as f:
-        f.write({"prompt": prompt, "refined_prompt": response, "image": "outputs/images/" + time})
+        f.write({"prompt": prompt, "refined_prompt": response, "image": "outputs/images/" + time + ".png"})
     return image
 
 # Create a Gradio Blocks app