changs

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from util.text_img import generate_image
+# from util.text_img import generate_image
 from util.v3d import generate_v3d, prep
 
 # Prepare the V3D model
@@ -7,19 +7,19 @@ model, clip_model, ae_model, device, num_frames, num_steps, rembg_session, outpu
 
 _TITLE = "Shoe Generator"
 with gr.Blocks(_TITLE) as ShoeGen:
-    with gr.Tab("Text to Image Generator"):
-        with gr.Row():
-            with gr.Column():
-                prompt = gr.Textbox(label="Enter a discription of a shoe")
-                # neg_prompt = gr.Textbox(label="Enter a negative prompt", value="low quality, watermark, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, body out of frame, blurry, bad anatomy, blurred, watermark, grainy, signature, cut off, draft, closed eyes, text, logo")
-                button_gen = gr.Button("Generate Image")
-            with gr.Column():
-                with gr.Tab("With Background"):
-                    image = gr.Image(label="Generated Image", show_download_button=True, show_label=False)
-                with gr.Tab("Without Background"): 
-                    image_nobg = gr.Image(label="Generated Image", show_download_button=True, show_label=False) 
+    # with gr.Tab("Text to Image Generator"):
+    #     with gr.Row():
+    #         with gr.Column():
+    #             prompt = gr.Textbox(label="Enter a discription of a shoe")
+    #             # neg_prompt = gr.Textbox(label="Enter a negative prompt", value="low quality, watermark, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, body out of frame, blurry, bad anatomy, blurred, watermark, grainy, signature, cut off, draft, closed eyes, text, logo")
+    #             button_gen = gr.Button("Generate Image")
+    #         with gr.Column():
+    #             with gr.Tab("With Background"):
+    #                 image = gr.Image(label="Generated Image", show_download_button=True, show_label=False)
+    #             with gr.Tab("Without Background"): 
+    #                 image_nobg = gr.Image(label="Generated Image", show_download_button=True, show_label=False) 
     
-        button_gen.click(generate_image, inputs=[prompt], outputs=[image, image_nobg])
+    #     button_gen.click(generate_image, inputs=[prompt], outputs=[image, image_nobg])
 
     with gr.Tab("Image to Video Generator (V3D)"):
         with gr.Row(equal_height=True):

requirements.txt

@@ -1,4 +1,3 @@
-# imports for image-to-V3D
 black==23.7.0
 chardet==5.1.0
 clip @ git+https://github.com/openai/CLIP.git
@@ -21,24 +20,24 @@ pudb>=2022.1.3
 pytorch-lightning==2.0.1
 pyyaml>=6.0.1
 scipy>=1.10.1
-# streamlit>=0.73.1
+streamlit>=0.73.1
 tensorboardx==2.6
 timm>=0.9.2
-# tokenizers==0.12.1
+tokenizers==0.12.1
 torch==2.0.1
 torchaudio>=2.0.2
 torchdata==0.6.1
 torchmetrics>=1.0.1
 torchvision>=0.15.2
 tqdm>=4.65.0
-# transformers==4.19.1
+transformers==4.19.1
 # triton==2.0.0
 urllib3<1.27,>=1.25.4
 wandb>=0.15.6
 webdataset>=0.2.33
 wheel>=0.41.0
 xformers>=0.0.20
-# streamlit-keyup==0.2.0
+streamlit-keyup==0.2.0
 mediapy
 tyro
 wget
@@ -46,12 +45,4 @@ rembg
 kiui
 spaces
 imageio==2.19.3
-imageio-ffmpeg==0.4.7
-
-# imports for text-to-image
-gradio
-diffusers==0.26.3
-transformers==4.38.1
-accelerate==0.27.2
-Python-IO
-huggingface-hub
+imageio-ffmpeg==0.4.7