Update app.py
Browse files
app.py
CHANGED
@@ -38,12 +38,12 @@ def chat_with_llava(message, history, system_message, max_tokens, temperature, t
|
|
38 |
demo = gr.Interface(
|
39 |
fn=chat_with_llava,
|
40 |
inputs=[
|
41 |
-
gr.
|
42 |
-
gr.
|
43 |
-
gr.
|
44 |
-
gr.
|
45 |
-
gr.
|
46 |
-
gr.
|
47 |
],
|
48 |
outputs=gr.outputs.Textbox(label="Response", placeholder="Model response will appear here..."),
|
49 |
title="LLAVA Model - Medical Image and Question",
|
|
|
38 |
demo = gr.Interface(
|
39 |
fn=chat_with_llava,
|
40 |
inputs=[
|
41 |
+
gr.Textbox(label="Message"),
|
42 |
+
gr.Image(label="Upload Medical Image"),
|
43 |
+
gr.Textbox(label="System message", default="You are a friendly Chatbot."),
|
44 |
+
gr.Slider(minimum=1, maximum=2048, default=512, step=1, label="Max new tokens"),
|
45 |
+
gr.Slider(minimum=0.1, maximum=4.0, default=0.7, step=0.1, label="Temperature"),
|
46 |
+
gr.Slider(minimum=0.1, maximum=1.0, default=0.95, step=0.05, label="Top-p (nucleus sampling)")
|
47 |
],
|
48 |
outputs=gr.outputs.Textbox(label="Response", placeholder="Model response will appear here..."),
|
49 |
title="LLAVA Model - Medical Image and Question",
|