Update app.py

app.py

@@ -47,6 +47,7 @@ def infer(prompt, n_samples, steps, scale, seed):
 
     response = safe_sd(prompt, int(n_samples), max(50,int(steps)), scale, seed, mode="text2img")
 
+    print(response.json()['output']['choices'][0]['image_base64'])
     data = json.load(BytesIO(response.content))
     if 'output' not in data:
         raise gr.Error("An error occurred.")