import gradio as gr import cv2 import torch import numpy as np from torchvision import transforms import torch from pytorch_lightning import seed_everything from torchvision.utils import save_image from model_lib.modules import MoMA_main_modal from model_lib.utils import parse_args import os os.environ["CUDA_VISIBLE_DEVICES"]="0" title = "MoMA" description = "This space has been migrated to Zero_GPU: https://huggingface.co/spaces/yizhezhu/MoMA_zeroGPU" def MoMA_demo(rgb, subject, prompt, strength, seed): seed = int(seed) if seed else 0 try: seed = int(seed) except ValueError: seed = 0 seed = seed if not seed == 0 else np.random.randint(0,1000) print(f"Seed: {seed}") with torch.no_grad(): generated_image = model.generate_images() return generated_image def inference(rgb, subject, prompt, strength, seed): result = MoMA_demo(rgb, subject, prompt, strength, seed) return result seed_everything(0) args = parse_args() #load MoMA from HuggingFace. Auto download model = MoMA_main_modal(args).to(args.device, dtype=torch.float16) gr.Interface( inference, [gr.Image(type="pil", label="Input RGB"), gr.Textbox(lines=1, label="subject"), gr.Textbox(lines=1, label="Prompt"), gr.Slider(minimum=0.2, maximum=1.2, step=0.1,label="Strength. Recommend: 1.0 for context editing; 0.4 for texture editing",value=1.0), gr.Textbox(lines=1, label="Seed. Use 0 for a random seed")], gr.Image(type="pil", label="Output"), title=title, description=description, examples=[["example_images/newImages/3.jpg",'car','A car in autumn with falling leaves.',1.0,"6"],["example_images/newImages/3.jpg",'car','A wooden sculpture of a car on a table.',0.4,"4"],["example_images/newImages/2.jpg",'car','A car on a city road with green trees and buildings.',1.0,"4"],["example_images/newImages/03.jpg",'cat','A cat at the Grand Canyon.',1.0,"2"],["example_images/newImages/02.jpg",'dog','A dog in a spring garden with flowers.',1.0,"6"],["example_images/newImages/1.jpeg",'bird','A bird in spring with flowers.',1.0,"1"],["example_images/newImages/17.jpg",'robot','A robot in autumn mountain and lake.',1,"5"]], allow_flagging='never' ).launch(debug=False)