Spaces:

yupikopi
/

test5

Sleeping

yupikopi commited on Apr 7, 2023

Commit

fc0fac4

1 Parent(s): 62062d9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,8 +24,8 @@ def sepia(inp):
     sepia_img = sepia_img / 255.0
     sepia_img = model.predict(sepia_img)
     sepia_img = sepia_img*255
-    #sepia_img = sepia_img.reshape(img_height, img_width, 3)
     sepia_img = Image.fromarray(sepia_img)
     return sepia_img
 demo = gr.Interface(fn=sepia, inputs=gr.inputs.Image(256,256),outputs="image").launch()

     sepia_img = sepia_img / 255.0
     sepia_img = model.predict(sepia_img)
     sepia_img = sepia_img*255
     sepia_img = Image.fromarray(sepia_img)
+    sepia_img = sepia_img.reshape(img_height, img_width, 3)
     return sepia_img
 demo = gr.Interface(fn=sepia, inputs=gr.inputs.Image(256,256),outputs="image").launch()