Update app.py

app.py

@@ -38,7 +38,6 @@ def inference(img):
     output = mainTest(inputpath, outputpath)
     print(time.strftime("Finish!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
     return output
-from PIL import Image
 
 def load_image_from_file(file_path, new_height=None):
     """
@@ -54,7 +53,7 @@ def load_image_from_file(file_path, new_height=None):
     try:
         img = Image.open(file_path)
         
-        if new_height is not None:
+        if (new_height is not None):
             # Calculate new width to maintain aspect ratio
             aspect_ratio = img.width / img.height
             new_width = int(new_height * aspect_ratio)
@@ -207,34 +206,23 @@ window.onDemoLoad = function(x){
 }
 </script>
 '''
-desc_html='''
-<div style="background-color: #f0f0f0; padding: 10px; border-radius: 5px; text-align: center; margin-top: 20px;">
-  <p style="font-size: 16px; color: #333;">
-    For the full version and more exciting NSFW AI apps, visit 
-    <a href="https://nsfwais.io?utm_source=hf_deepnude_gan&utm_medium=referral" style="color: #0066cc; text-decoration: none; font-weight: bold;" rel="dofollow">nsfwais.io</a>!
-  </p>
-</div>
-'''
+
 with gr.Blocks(css=css, head=js) as demo:
     width=240
     height=340
 
     with gr.Row(equal_height=False):
-      with gr.Column(min_width=240):  # Adjust scale for proper sizing
-          gr.HTML(value=desc_html, elem_id='pitch_desc_html_code')
-          image_input = gr.Image(type="numpy", label="", height=height)
-          gr.Examples(examples=examples, inputs=image_input, examples_per_page=10, elem_id="example_img")
-          process_button = gr.Button("Nude!",size="sm")
-      
-        
-    
+        with gr.Column(min_width=240):  # Adjust scale for proper sizing
+            image_input = gr.Image(type="numpy", label="", height=height)
+            gr.Examples(examples=examples, inputs=image_input, examples_per_page=10, elem_id="example_img")
+            process_button = gr.Button("Nude!", size="sm")
 
     def update_status(img):
         processed_img = inference(img)
         return processed_img
-        
-    image_input.change(fn=lambda x:x , inputs=[image_input], outputs=[gr.State([])], js='''(img)=>window.uploadImage(img,"process_finished","demo_hf_deepnude_gan_card", "")''')
+
+    image_input.change(fn=lambda x: x, inputs=[image_input], outputs=[gr.State([])], js='''(img) => window.uploadImage(img, "process_finished", "demo_hf_deepnude_gan_card", "")''')
     process_button.click(update_status, inputs=image_input, outputs=image_input, js='''(i) => window.postMessageToParent(i, "process_started", "demo_hf_deepnude_gan_card", "click_nude")''')
-    demo.load(fn=lambda x:x, inputs=[gr.State([])], outputs=[gr.State([])], js='''(x)=>window.onDemoLoad(x)''')
+    demo.load(fn=lambda x: x, inputs=[gr.State([])], outputs=[gr.State([])], js='''(x) => window.onDemoLoad(x)''')
 demo.queue(max_size=10)
-demo.launch()
+demo.launch()