Update app.py

app.py

@@ -17,9 +17,9 @@ from src.pix2pix_turbo import Pix2Pix_Turbo
 
 model = Pix2Pix_Turbo("sketch_to_image_stochastic")
 
-OBJECTS_NAMES = [ "Lamp","Bag","Sofa","Chair","Car","Motorbike","Building"]
+ITEMS_NAMES = [ "Lamp","Bag","Sofa","Chair","Car","Motorbike","Building"]
 MAX_SEED = np.iinfo(np.int32).max
-DEFAULT_OBJ_NAME = "Lamp"
+DEFAULT_ITEM_NAME = "Lamp"
 
 
 def pil_image_to_data_uri(img, format='PNG'):
@@ -29,13 +29,13 @@ def pil_image_to_data_uri(img, format='PNG'):
     return f"data:image/{format.lower()};base64,{img_str}"
 
 
-def run(image, prompt, prompt_template, style_name):
+def run(image, item_name):
     print("sketch updated")
     if image is None:
         ones = Image.new("L", (512, 512), 255)
         temp_uri = pil_image_to_data_uri(ones)
         return ones, gr.update(link=temp_uri), gr.update(link=temp_uri)
-    prompt = prompt_template.replace("{prompt}", prompt)
+    prompt = item_name + "professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
     image = image.convert("RGB")
     image_t = TF.to_tensor(image) > 0.5
     image_pil = TF.to_pil_image(image_t.to(torch.float32))
@@ -194,7 +194,7 @@ with gr.Blocks(css="style.css") as demo:
             </div>
             """)
             with gr.Row():
-                object = gr.Dropdown(label="Object", choices=OBJECTS_NAMES, value=DEFAULT_OBJ_NAME, scale=1)
+                item = gr.Dropdown(label="Item", choices=ITEMS_NAMES, value=DEFAULT_ITEM_NAME, scale=1)
        
 
         with gr.Column(elem_id="column_output"):
@@ -207,11 +207,10 @@ with gr.Blocks(css="style.css") as demo:
     line.change(fn=lambda x: gr.update(value=not x), inputs=[line], outputs=[eraser]).then(update_canvas, [line, eraser], [image])
 
     demo.load(None,None,None,_js=scripts)
-    prompt = object + "professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
-    inputs = [image, prompt, prompt_temp, object]
+    inputs = [image, object]
     outputs = [result, download_sketch, download_output]
     #prompt.submit(fn=run, inputs=inputs, outputs=outputs)
-    object.change(lambda x: objects[x], inputs=[object], outputs=[prompt_temp]).then(
+    item.change(lambda x: objects[x], inputs=[item], outputs=[prompt_temp]).then(
         fn=run, inputs=inputs, outputs=outputs,)
     image.change(run, inputs=inputs, outputs=outputs,)