File size: 807 Bytes
ab0f9b3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import gradio as gr
from zs_audio import classify_audio
from zs_image import classify_image
audio_interface = gr.Interface(
fn=classify_audio,
inputs=gr.Audio(sources=["microphone", "upload"], type="filepath"),
outputs=gr.Label(),
title="Zero-Shot Audio Classification",
description="Classify audio into predefined categories without prior training.",
allow_flagging="never",
)
image_interface = gr.Interface(
fn=classify_image,
inputs=gr.Image(type="filepath"),
outputs=gr.Label(),
title="Zero-Shot Image Classification",
description="Classify an image into predefined categories using CLIP.",
allow_flagging="never",
)
app = gr.TabbedInterface(
[audio_interface, image_interface],
["Audio Classification", "Image Classification"]
)
app.launch()
|