Update app.py

app.py

@@ -23,7 +23,6 @@ Original file is located at
 # Импорт нужных библиотек
 import os
 import sys
-from tqdm import tqdm
 import torch
 import torch.nn as nn
 import torch.optim as optim
@@ -31,19 +30,17 @@ from torchvision import transforms
 from torchvision.utils import save_image
 from PIL import Image
 import numpy as np
-import matplotlib.pyplot as plt
+import gradio as gr
+import subprocess
+import gdown
 
 # Настройка устройства
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-import os
-import subprocess
-
 if not os.path.exists("stylegan2-pytorch"):
     subprocess.run(["git", "clone", "https://github.com/rosinality/stylegan2-pytorch.git"])
 os.chdir("stylegan2-pytorch")
 
-import gdown
 gdown.download('https://drive.google.com/uc?id=1EM87UquaoQmk17Q8d5kYIAHqu0dkYqdT')
 gdown.download('https://drive.google.com/uc?id=1N0MZSqPRJpLfP4mFQCS14ikrVSe8vQlL')
 sys.path.append("/home/user/app/stylegan2-pytorch")
@@ -56,26 +53,13 @@ state_dict = torch.load('stylegan2-ffhq-config-f.pt', map_location=device)
 f_generator.load_state_dict(state_dict['g_ema'])
 f_generator.eval()
 
-g_generator = Generator(size=1024, style_dim=latent_dim, n_mlp=8).to(device)
-g_generator.load_state_dict(state_dict['g_ema'])
-
-# Загрузка модели CLIP
-import clip
-clip_model, preprocess = clip.load("ViT-B/32", device=device)
-
-latent_dim=512
-batch_size=4
-
-"""## 6. Готовые пресеты"""
-
-# @title Загрузка пресетов
+# Загрузка пресетов
 os.makedirs("/content/presets", exist_ok=True)
 
 gdown.download('https://drive.google.com/uc?id=1trcBvlz7jeBRLNeCyNVCXE4esW25GPaZ', '/content/presets/sketch.pth')
 gdown.download('https://drive.google.com/uc?id=1N4C-aTwxeOamZX2GeEElppsMv-ALKojL', '/content/presets/modigliani.pth')
 gdown.download('https://drive.google.com/uc?id=1VZHEalFyEFGWIaHei98f9XPyHHvMBp6J', '/content/presets/werewolf.pth')
 
-# @title Генерация примеров из пресета
 # Загрузка генератора из файла
 def load_model(file_path, latent_dim=512, size=1024):
 
@@ -97,11 +81,6 @@ model_paths = {
     "Human -> Werewolf": "/content/presets/werewolf.pth"
 }
 
-
-"""## 8. Веб-демо"""
-
-import gradio as gr
-
 # Функция обработки
 def generate(model_name):
     model_path = model_paths[model_name]