scribbyotx's picture
Update app.py
8a7f664 verified
raw
history blame contribute delete
No virus
560 Bytes
import gradio as gr
import replicate
# Authenticate with Replicate
replicate.Client(api_token="r8_bf0m2SnfXi6m1YcVGuAxekbghbxVfE02qgTqV")
# Define a function to use the model
def generate_image(prompt):
model = replicate.models.get("xarty8932/dream")
version = model.versions.get("5e3c45aa9c9896f86634175309490225e5a379a6a81c39abbf55eab2cd16b657")
output = version.predict(prompt=prompt)
return output
# Create a Gradio interface
iface = gr.Interface(fn=generate_image, inputs="text", outputs="image")
# Launch the interface
iface.launch()