Update app.py

app.py

@@ -15,11 +15,10 @@ import glob
 import numpy as np
 from basicsr.utils import imwrite
 from gfpgan import GFPGANer
-
 #os.system("pip freeze")
 #os.system("wget https://github.com/TencentARC/GFPGAN/releases/download/v0.2.0/GFPGANCleanv1-NoCE-C2.pth -P .")
 import random
-import gradio as gr
+
 from PIL import Image
 import torch
 # torch.hub.download_url_to_file('https://upload.wikimedia.org/wikipedia/commons/thumb/a/ab/Abraham_Lincoln_O-77_matte_collodion_print.jpg/1024px-Abraham_Lincoln_O-77_matte_collodion_print.jpg', 'lincoln.jpg')
@@ -88,16 +87,16 @@ for img_path in img_list:
 
 
 
- 
 
-def inference(img):
+
+def inference():
     os.system("ls ./_output")
-    os.system("ls ./_output/restored_imgs")
-    input_img = cv2.imread("./_output/restored_imgs/imagem-0001.png" , cv2.IMREAD_COLOR)
-   
+    os.system("ls ./_output/cmp")
+    input_img = cv2.imread("./_output/cmp/imagem-0001.png" , cv2.IMREAD_COLOR)
+    st.image(input_img)
     
     #return Image.fromarray(restored_faces[0][:,:,::-1])
-    return input_img
+    
 
 title = "Melhoria de imagens"
 
@@ -105,18 +104,6 @@ os.system("ls")
 description = "Sistema para automação。"
 
 article = "<p style='text-align: center'><a href='https://huggingface.co/spaces/akhaliq/GFPGAN/' target='_blank'>clone from akhaliq@huggingface with little change</a> | <a href='https://github.com/TencentARC/GFPGAN' target='_blank'>GFPGAN Github Repo</a></p><center><img src='https://visitor-badge.glitch.me/badge?page_id=akhaliq_GFPGAN' alt='visitor badge'></center>"
-st.button('OI')
-gr.Interface(
-    inference, 
-    [gr.inputs.Image(type="filepath", label="Input")], 
-    gr.outputs.Image(type="pil", label="Output"),
-    title=title,
-    description=description
-    #,
-   #examples=[  
-    
-    #['edison.jpg'],
-    #['pessoa3.jpg']
-    #]
-    ).launch(enable_queue=True,cache_examples=False,share=True)
+st.button('OI',on_click=inference)
+