Update app.py

app.py

@@ -39,21 +39,6 @@ model.renderer.set_chunk_size(131072)
 model.to(device)
 
 rembg_session = rembg.new_session()
-my_aws_access_key_id = os.getenv("ACCESS")
-my_aws_secret_access_key = os.getenv("SECRET")
-bedrock_runtime = boto3.client(service_name='bedrock-runtime', aws_access_key_id = my_aws_access_key_id, aws_secret_access_key = my_aws_secret_access_key, region_name='us-east-1')
-
-def generate_image_from_text(pos_prompt):
-    # bedrock_runtime = boto3.client(region_name = 'us-east-1', service_name='bedrock-runtime')
-    parameters = {'text_prompts': [{'text': pos_prompt , 'weight':1},
-                                    {'text': """Blurry, out of frame, out of focus, Detailed, dull, duplicate, bad quality, low resolution, cropped""", 'weight': -1}],
-                  'cfg_scale': 7, 'seed': 0, 'samples': 1}
-    request_body = json.dumps(parameters)
-    response = bedrock_runtime.invoke_model(body=request_body,modelId = 'stability.stable-diffusion-xl-v1')
-    response_body = json.loads(response.get('body').read())
-    base64_image_data = base64.b64decode(response_body['artifacts'][0]['base64'])
-    
-    return Image.open(io.BytesIO(base64_image_data))
 
 def check_input_image(input_image):
     if input_image is None:
@@ -109,10 +94,10 @@ with gr.Blocks() as demo:
     with gr.Row(variant="panel"):
         with gr.Column():
             with gr.Row():
-                text_prompt = gr.Textbox(
-                    label="Text Prompt",
-                    placeholder="Enter a text prompt for image generation"
-                )
+                # text_prompt = gr.Textbox(
+                #     label="Text Prompt",
+                #     placeholder="Enter a text prompt for image generation"
+                # )
                 input_image = gr.Image(
                     label="Generated Image",
                     image_mode="RGBA",
@@ -156,21 +141,21 @@ with gr.Blocks() as demo:
                     interactive=False,
                 )
                 gr.Markdown("Note: The model shown here has a darker appearance. Download to get correct results.")
-    # with gr.Row(variant="panel"):
-    #     gr.Examples(
-    #         examples=[
-    #             os.path.join("examples", img_name) for img_name in sorted(os.listdir("examples"))
-    #         ],
-    #         inputs=[text_prompt],
-    #         outputs=[processed_image, output_model_obj, output_model_glb],
-    #         cache_examples=True,
-    #         fn=partial(run_example, do_remove_background=True, foreground_ratio=0.85, mc_resolution=256),
-    #         label="Examples",
-    #         examples_per_page=20
-    #     )
+    with gr.Row(variant="panel"):
+        gr.Examples(
+            examples=[
+                os.path.join("examples", img_name) for img_name in sorted(os.listdir("examples"))
+            ],
+            inputs=[text_prompt],
+            outputs=[processed_image, output_model_obj, output_model_glb],
+            cache_examples=True,
+            fn=partial(run_example, do_remove_background=True, foreground_ratio=0.85, mc_resolution=256),
+            label="Examples",
+            examples_per_page=20
+        )
     submit.click(fn=check_input_image, inputs=[text_prompt]).success(
         fn=run_example,
-        inputs=[text_prompt, do_remove_background, foreground_ratio, mc_resolution],
+        inputs=[input_image, do_remove_background, foreground_ratio, mc_resolution],
         outputs=[processed_image, output_model_obj, output_model_glb],
     )