Geo-Painting / app.py
DavidTamayo's picture
Update to use pillow image instead np array image
1b68d22
import gradio as gr
from geo_painting import GeoPainting
geo_painting = GeoPainting()
def generate_image(input_promp, control_image):
return geo_painting.generate_painting(input_promp, control_image)
input_promp = gr.Textbox(label="Input promp")
control_image = gr.Image(label="Select a control image")
output_image = gr.Image(label="Generated geospatial image")
iface = gr.Interface(generate_image, [input_promp, control_image], output_image)
iface.launch()