Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 from PIL import Image
 import torch
-
+os.system("pip install gradio=2.5.3")
 import gradio as gr
 import torch
 torch.backends.cudnn.benchmark = True
@@ -16,15 +16,16 @@ from torch.nn import functional as F
 from tqdm import tqdm
 import lpips
 from model import *
-os.makedirs('models', exist_ok=True)
-os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
-os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
+
 
 from e4e_projection import projection as e4e_projection
 
 from copy import deepcopy
 import imageio
 
+os.makedirs('models', exist_ok=True)
+os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
+os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
 
 os.system("wget http://dlib.net/files/shape_predictor_68_face_landmarks.dat.bz2")
 os.system("bzip2 -dk shape_predictor_68_face_landmarks.dat.bz2")
@@ -150,4 +151,4 @@ description = "Gradio Demo for JoJoGAN: One Shot Face Stylization. To use it, si
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2112.11641' target='_blank'>JoJoGAN: One Shot Face Stylization</a>| <a href='https://github.com/mchong6/JoJoGAN' target='_blank'>Github Repo Pytorch</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=akhaliq_jojogan' alt='visitor badge'></center>"
 
 examples=[['mona.png','Jinx']]
-gr.Interface(inference, [gr.inputs.Image(type="filepath"),gr.inputs.Dropdown(choices=['JoJo', 'Disney','Jinx','Caitlyn','Yasuho','Arcane Multi','Art','Spider-Verse'], type="value", default='JoJo', label="Model")], gr.outputs.Image(type="file"),title=title,description=description,article=article,allow_flagging="never",examples=examples,allow_screenshot=False).launch(enable_queue=True)
+gr.Interface(inference, [gr.inputs.Image(type="filepath"),gr.inputs.Dropdown(choices=['JoJo', 'Disney','Jinx','Caitlyn','Yasuho','Arcane Multi','Art','Spider-Verse'], type="value", default='JoJo', label="Model")], gr.outputs.Image(type="file"),title=title,description=description,article=article,allow_flagging=False,examples=examples,allow_screenshot=False,enable_queue=True).launch()