fluffyvision / app.py
j4y1337's picture
gpu preparation
2e31314 verified
import gradio as gr
import predictLLaVa
images = gr.Files(label="Image(s)", file_types=list(".png .jpg .jpeg".split()))
inputText = gr.Textbox(lines=1, label="English Instruction:")
languageDropdown = gr.Dropdown(
choices=["en", "de", "fr", "es", "it", "nl", "ru"],
label="Output in:",
multiselect=False,
value="en"
)
gr.Interface(
fn=predictLLaVa.process_single_image,
inputs=[inputText, images, languageDropdown],
outputs=gr.Textbox(label="Output", show_copy_button=True),
title="Fluffyvision",
description="Generate a text description of an image based on an instruction.",
allow_flagging="never"
).launch(auth=predictLLaVa.check_auth, auth_message="Please enter your username and password.")