Update app.py

app.py

@@ -161,12 +161,12 @@ def runModel(content, style):
     if os.path.exists("0.png"):
         os.remove("0.png")
     gltf = GLTF2().load(content)
-    gltf.convert_images(ImageFormat.FILE,"/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/", override=True)
+    gltf.convert_images(ImageFormat.FILE,"/tmp/", override=True)
     gltf.images[0].uri
     gltf_reinhard = GLTF2().load(content)
     gltf_lhm = GLTF2().load(content)
     gltf_pccm = GLTF2().load(content)
-    ori_image=PIL.Image.open('/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/0.png')
+    ori_image=PIL.Image.open('/tmp/0.png')
     ori_image=ori_image.convert('RGB')
     content_img = np.array(ori_image)[:, :, :3]
     style_img = np.array(style)[:, :, :3]
@@ -176,15 +176,15 @@ def runModel(content, style):
     image_reinhard = PIL.Image.fromarray(output_reinhard,'RGB')
     image_lhm = PIL.Image.fromarray(output_lhm,'RGB')
     image_pccm = PIL.Image.fromarray(output_pccm,'RGB')
-    image_reinhard=image_reinhard.save('/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/reinhard.png')
-    image_lhm=image_lhm.save('/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/lhm.png')
-    image_pccm=image_pccm.save('/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/pccm.png')
+    image_reinhard=image_reinhard.save('/tmp/reinhard.png')
+    image_lhm=image_lhm.save('/tmp/lhm.png')
+    image_pccm=image_pccm.save('/tmp/pccm.png')
     image1=Image()
     image2=Image()
     image3=Image()
-    image1.uri='/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/reinhard.png'
-    image2.uri='/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/lhm.png'
-    image3.uri='/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/pccm.png'
+    image1.uri='/tmp/reinhard.png'
+    image2.uri='/tmp/lhm.png'
+    image3.uri='/tmp/pccm.png'
     gltf_reinhard.images[0]=image1
     gltf_lhm.images[0]=image2
     gltf_pccm.images[0]=image3
@@ -197,10 +197,10 @@ def runModel(content, style):
     gltf_lhm.images[0].name 
     gltf_pccm.images[0].uri 
     gltf_pccm.images[0].name
-    gltf_pccm.save('/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/pccm.glb')
-    gltf_lhm.save('/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/lhm.glb')
-    gltf_reinhard.save('/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/reinhard.glb')
-    return '/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/pccm.glb','/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/lhm.glb', '/tmp/gradio/d81e337d5bd4f65969579d50a84697257477c1d6/output/reinhard.glb'
+    gltf_pccm.save('/tmp/pccm.glb')
+    gltf_lhm.save('/tmp/lhm.glb')
+    gltf_reinhard.save('/tmp/reinhard.glb')
+    return '/tmp/pccm.glb','/tmp/lhm.glb', '/tmp/reinhard.glb'
 
 demo = gr.Interface(runModel, [gr.Model3D(),gr.Image(type ='pil')], outputs = [gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="Reinhard"), gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="lhm"), gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="pccm")], )