Update app.py

app.py

@@ -2,9 +2,14 @@ import streamlit as st
 import torch
 from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
 from PIL import Image
+from transformers import AutoProcessor, AutoModelForImageTextToText
+
+
 
 # Load the model and processor
 model_id = "brucewayne0459/paligemma_derm"
+processor_chat = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
+model_chat = AutoModelForImageTextToText.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")  
 processor = AutoProcessor.from_pretrained(model_id)
 model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
 model.eval()
@@ -43,10 +48,11 @@ col1, col2 = st.columns([3, 2])
 with col1:
     # File uploader for image
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+    prompt = "Identify the skin condition?"
     
     # Text input for prompt
-    input_text = st.text_input("Enter your prompt:", "Identify the skin condition?")
-
+    input_text = st.text_input("Enter your prompt:", prompt)
+    
 with col2:
     # Display uploaded image (if any)
     if uploaded_file: