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('runwayml/stable-diffusion-v1-5', 'huggingface')

interface.launch(share = True)