update

README.md

@@ -1,5 +1,5 @@
 ---
-title: GFPGAN
+title: GFPGAN-ZEROGPU
 emoji: 😁
 colorFrom: yellow
 colorTo: green

app.py

@@ -6,6 +6,7 @@ import torch
 from basicsr.archs.srvgg_arch import SRVGGNetCompact
 from gfpgan.utils import GFPGANer
 from realesrgan.utils import RealESRGANer
+import spaces
 
 os.system("pip freeze")
 # download weights
@@ -45,6 +46,7 @@ os.makedirs('output', exist_ok=True)
 
 
 # def inference(img, version, scale, weight):
+@spaces.GPU(enable_queue=True)
 def inference(img, version, scale):
     # weight /= 100
     print(img, version, scale)
@@ -113,21 +115,17 @@ def inference(img, version, scale):
 
 
 title = "GFPGAN: Practical Face Restoration Algorithm"
-description = r"""Gradio demo for <a href='https://github.com/TencentARC/GFPGAN' target='_blank'><b>GFPGAN: Towards Real-World Blind Face Restoration with Generative Facial Prior</b></a>.<br>
+description = r"""GFPGAN On ZeroGPU : Towards Real-World Blind Face Restoration with Generative Facial Prior</b></a>.<br>
 It can be used to restore your **old photos** or improve **AI-generated faces**.<br>
 To use it, simply upload your image.<br>
-If GFPGAN is helpful, please help to ⭐ the <a href='https://github.com/TencentARC/GFPGAN' target='_blank'>Github Repo</a> and recommend it to your friends 😊
 """
 article = r"""
+ This is a clone of the original <a href='https://huggingface.co/spaces/Xintao/GFPGAN' target='_blank'>GFPGAN demo</a> with the following changes:
+- The original demo was modified to use the GPU for inference.
+- changed touch version to v2.0.0
 
-[![download](https://img.shields.io/github/downloads/TencentARC/GFPGAN/total.svg)](https://github.com/TencentARC/GFPGAN/releases)
-[![GitHub Stars](https://img.shields.io/github/stars/TencentARC/GFPGAN?style=social)](https://github.com/TencentARC/GFPGAN)
-[![arXiv](https://img.shields.io/badge/arXiv-Paper-<COLOR>.svg)](https://arxiv.org/abs/2101.04061)
+Please follow me on [Linkedin](https://www.linkedin.com/in/saifturzo3/).
 
-If you have any question, please email 📧 `[email protected]` or `[email protected]`.
-
-<center><img src='https://visitor-badge.glitch.me/badge?page_id=akhaliq_GFPGAN' alt='visitor badge'></center>
-<center><img src='https://visitor-badge.glitch.me/badge?page_id=Gradio_Xintao_GFPGAN' alt='visitor badge'></center>
 """
 demo = gr.Interface(
     inference, [
@@ -147,4 +145,4 @@ demo = gr.Interface(
     #           ['10045.png', 'v1.4', 2, 50]]).launch()
     examples=[['AI-generate.jpg', 'v1.4', 2], ['lincoln.jpg', 'v1.4', 2], ['Blake_Lively.jpg', 'v1.4', 2],
               ['10045.png', 'v1.4', 2]])
-demo.queue().launch()
+demo.queue(max_size=50).launch()

requirements.txt

@@ -1,4 +1,4 @@
-torch>=1.7
+torch==2.0.0
 basicsr>=1.4.2
 facexlib>=0.2.5
 gfpgan>=1.3.7