img-to-music / app.py
fffiloni's picture
Update app.py
ba26d1c
raw
history blame
841 Bytes
import gradio as gr
import os
img_to_text = gr.Blocks.load(name="spaces/pharma/CLIP-Interrogator")
text_to_music = gr.Interface.load("spaces/Mubert/Text-to-Music")
def get_prompts(uploaded_image):
prompt = img_to_text(uploaded_image, fn_index=1)[0]
print(prompt)
music_result = get_music(prompt)
return music_result
def get_music(prompt):
email = "blabla@mail.com"
duration = 30
result = text_to_music(email, prompt, duration)
output = os.path.join(result)
print(output)
return prompt
with gr.Blocks() as demo:
with gr.Row():
with gr.Column():
input_img = gr.Image(type="filepath")
generate = gr.Button("Generate Music from Image")
with gr.Column():
music_output = gr.Textbox(label="Result")
generate.click(get_prompts, inputs=[input_img], outputs=[music_output])
demo.launch()