Update app.py

app.py

@@ -3,7 +3,6 @@ from transformers import pipeline
 import multiprocessing
 import torch
 import os
-from google.colab import output, drive
 from psutil import virtual_memory
 from pathlib import Path
 import random
@@ -19,8 +18,18 @@ from pathlib import Path
 import csv
 import logging
 import requests
+from rudalle.pipelines import generate_images, show, super_resolution, cherry_pick_by_ruclip
+from rudalle import get_rudalle_model, get_tokenizer, get_vae, get_realesrgan
+from rudalle.utils import seed_everything
+from ruclip import load as get_ruclip
 
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+# prepare models:
+device = 'cuda'
+model = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
+tokenizer = get_tokenizer()
+vae = get_vae(dwt=True).to(device)
+
+#device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 
 pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")