Reset

Rodin.py

@@ -74,7 +74,6 @@ def crop_image(image, type):
         new_image.paste(cropped_image, (i * 360, 0))
     return new_image
 
-
 # Perform Rodin mesh operation
 def rodin_mesh(prompt, group_uuid, settings, images, name, token):
     images = [convert_base64_to_binary(img) for img in images]
@@ -95,7 +94,6 @@ def rodin_mesh(prompt, group_uuid, settings, images, name, token):
     response = requests.post(f"{BASE_URL}/task/rodin_mesh", data=m, headers=headers)
     return response
 
-
 # Convert base64 to binary since the result from `rodin_preprocess_image` is encoded with base64
 def convert_base64_to_binary(base64_string):
     if ',' in base64_string:

app.py

@@ -3,10 +3,11 @@ os.system('pip uninstall -y gradio_fake3d')
 os.system('pip install gradio_fake3d-0.0.3-py3-none-any.whl')
 
 import gradio as gr
+import spaces
 import re
 from gradio_fake3d import Fake3D
 from PIL import Image
-from Rodin import Generator, crop_image, log, convert_base64_to_binary
+from Rodin import Generator, crop_image, log
 from constant import *
 
 generator = Generator(USER, PASSWORD, TOKEN)
@@ -54,39 +55,11 @@ html_content = """
 </div>
 """
 
-max_height = 256
-
-css = f"""
-#right_col_group_row_gright > div, #fake3d, #right_col_group_row_gleft > div, #block_3d {{
-    flex-grow: 1;
-}}
-#elem_block_image img {{
-    max-height:{max_height}px;
-    object-fit: contain;
-}}
-#elem_block_image_crop img {{
-    max-height:{max_height}px;
-    background: repeating-linear-gradient(
-        -45deg,
-        transparent 0px,
-        transparent 20px,
-        #f0f0f0 20px,
-        #f0f0f0 40px
-    )
-}}
-label {{
-    user-select: none
-}}
-#elem_block_download_mesh {{
-    flex-grow: 1
-}}
-#elem_block_download_mesh div {{
-    min-height: auto
-}}
-#elem_button_draw {{
-    flex-grow: 1
-}}
-"""
+options = [
+    "Rodin Gen-1(0525)",
+    "OpenClay(600M) - Coming soon",
+    "OpenClay(200M) - Coming soon"
+]
 
 example = [
     ["assets/00.png"],
@@ -133,7 +106,7 @@ def handle_prompt_change(prompt):
 
 def clear_task(task_input=None):
     """_summary_
-    [cache_task_uuid, block_prompt, block_prompt_hint, fake3d, block_3d]
+    [cache_task_uuid, block_prompt, block_prompt_hint, fake3d]
     """
     log("INFO", "Clearing task...")
     return "", "", "", [], "assets/white_image.png"
@@ -159,12 +132,6 @@ def crop_image_normal(image):
 
 with gr.Blocks() as demo:
     gr.HTML(html_content)
-
-    cache_task_uuid = gr.Text(value="", visible=False)
-    cache_raw_image = gr.Image(visible=False, type="pil")
-    cacha_empty = gr.Text(visible=False)
-    cache_image_base64 = gr.Text(visible=False)
-
     with gr.Row():
         with gr.Column():
             block_image = gr.Image(height=256, image_mode="RGB", sources="upload", elem_classes="elem_imageupload", type="filepath")