import os
import gradio as gr
import openai
openai.api_key = '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' # your api key
openai.Model.list()

prompt = "joe mama"

openai.Image.create(
  prompt=prompt
  n=2,
  size="1024x1024"
)