main.py

from fastapi import FastAPI
import gradio as gr
import torch
from diffusers import DiffusionPipeline
import uvicorn

description = "Image generation with GPT-2"
title = "IMAGE GENERATION MACHINE"

interface = gr.Interface.load('Lykon/DreamShaper', 'huggingface')

interface.launch(share = True)