Himanshu-AT commited on
Commit
e0123d5
·
1 Parent(s): c3e1273

export jpg

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -69,7 +69,12 @@ def infer(edit_images, prompt, seed=42, randomize_seed=False, width=1024, height
69
  num_inference_steps=num_inference_steps,
70
  generator=torch.Generator("cpu").manual_seed(seed)
71
  ).images[0]
72
- return image, seed
 
 
 
 
 
73
 
74
  examples = [
75
  "photography of a young woman, accent lighting, (front view:1.4), "
 
69
  num_inference_steps=num_inference_steps,
70
  generator=torch.Generator("cpu").manual_seed(seed)
71
  ).images[0]
72
+
73
+ output_image_jpg = image.convert("RGB")
74
+ output_image_jpg.save("output.jpg", "JPEG")
75
+
76
+ return output_image_jpg, seed
77
+ # return image, seed
78
 
79
  examples = [
80
  "photography of a young woman, accent lighting, (front view:1.4), "