Upload 2 files

app.py

@@ -1,37 +1,27 @@
 import sys
 import os
-import re
-from typing import List, Optional, Tuple, Union
 
-import click
+os.system("https://github.com/dunbar12138/pix2pix3D.git")
+sys.path.append("pix2pix3D")
+
+from typing import List, Optional, Tuple, Union
 import dnnlib
 import numpy as np
 import PIL.Image
 import torch
 from tqdm import tqdm
 
-
 import legacy
 from camera_utils import LookAtPoseSampler
 from huggingface_hub import hf_hub_download
-
 from matplotlib import pyplot as plt
-
 from pathlib import Path
-
-import json
 import gradio as gr
-
 from training.utils import color_mask, color_list
 import plotly.graph_objects as go
 from tqdm import tqdm
-
 import imageio
-
-import argparse
-
 import trimesh
-import pyrender
 import mcubes
 
 os.environ["PYOPENGL_PLATFORM"] = "egl"
@@ -198,11 +188,13 @@ def get_all(cfg,input,truncation_psi,mesh_resolution,random_seed,fps,num_frames)
         fig_mesh=return_plot_go(mesh_trimesh)
         return fig_mesh,image_color,image_seg,video,video_label
 
-markdown=f'''
-  # 3D-aware Conditional Image Synthesis
-  
+title="3D-aware Conditional Image Synthesis"
+desc=f'''
+
   [Arxiv:  "3D-aware Conditional Image Synthesis".](https://arxiv.org/abs/2302.08509)
+
   [Project Page.](https://www.cs.cmu.edu/~pix2pix3D/)
+
   [For the official implementation.](https://github.com/dunbar12138/pix2pix3D)
 
   ### Future Work based on interest
@@ -214,39 +206,39 @@ markdown=f'''
   The process can take long time.Especially ,To generate videos and the time of process depends the number of frames,Mesh Resolution and current compiler device.
   
 '''
-
-
-with gr.Blocks() as demo:
-    gr.Markdown(markdown)
-    with gr.Row():
-      with gr.Column():
-        input=gr.Image(type="filepath",shape=(512, 512))
-      with gr.Column():
-        cfg=gr.Dropdown(choices=["seg2cat"],label="Choose Model",value="seg2cat")
-        truncation_psi = gr.Slider( minimum=0, maximum=2,label='Truncation PSI',value=1)
-        mesh_resolution = gr.Slider( minimum=32, maximum=512,label='Mesh Resolution',value=32)
-        random_seed = gr.Slider( minimum=0, maximum=2**16,label='Seed',value=128)
-        fps = gr.Slider( minimum=10, maximum=120,label='FPS',value=30)
-        num_frames = gr.Slider( minimum=10, maximum=120,label='The Number of Frames',value=30)
-
-    with gr.Row():
-      btn = gr.Button(value="Generate")
-
-    with gr.Row():
-      with gr.Column():
-            image_color=gr.Image(type="pil",shape=(256,256))
-      with gr.Column():
-            image_label=gr.Image(type="pil",shape=(256,256))
-    with gr.Row():
-          mesh = gr.Plot()
-    with  gr.Row():
-        with gr.Column():
-            video_color=gr.Video()
-        with gr.Column():
-            video_label=gr.Video()
-
-
-
-    btn.click(get_all, [cfg,input,truncation_psi,mesh_resolution,random_seed,fps,num_frames],[ mesh,image_color,image_label,video_color,video_label])
-
-demo.launch(debug=True,share=True)
+demo_inputs=[
+    gr.Dropdown(choices=["seg2cat"],label="Choose Model",value="seg2cat"),
+    gr.Image(type="filepath",shape=(512, 512),label="Mask"),
+    gr.Slider( minimum=0, maximum=2,label='Truncation PSI',value=1),
+    gr.Slider( minimum=32, maximum=512,label='Mesh Resolution',value=32),
+    gr.Slider( minimum=0, maximum=2**16,label='Seed',value=128),
+    gr.Slider( minimum=10, maximum=120,label='FPS',value=30),
+    gr.Slider( minimum=10, maximum=120,label='The Number of Frames',value=30),
+
+]
+demo_outputs=[
+    gr.Plot(),
+    gr.Image(type="pil",shape=(256,256)),
+    gr.Image(type="pil",shape=(256,256)),
+    gr.Video(),
+    gr.Video()
+
+]
+examples = [
+    ["seg2cat", "example_input.png", 1, 32, 128, 30, 30],
+
+]
+    
+
+demo_app = gr.Interface(
+    fn=get_all,
+    inputs=demo_inputs,
+    outputs=demo_outputs,
+    cache_examples=True,
+    title=title,
+    theme="huggingface",
+    description=desc,
+    examples=examples,
+    cache_examples=True,
+)
+demo_app.launch(debug=True, enable_queue=True)