Update app.py

app.py

@@ -34,8 +34,8 @@ st.markdown(
 )
 
 # Streamlit app title and instructions
-st.title("Skin Condition Identifier")
-st.write("Upload an image and provide a custom prompt to identify the skin condition.")
+st.title("VisionDerm")
+st.write("Upload an image to identify the skin condition.")
 
 # Column layout for input and display
 col1, col2 = st.columns([3, 2])
@@ -72,12 +72,21 @@ with col2:
                 with torch.no_grad():
                     outputs = model.generate(**inputs, max_new_tokens=default_max_tokens)
 
-                # Decode output
+                # Decode output and remove the prompt text from the response
                 decoded_output = processor.decode(outputs[0], skip_special_tokens=True)
-
+                
+                # Remove the prompt text from the model output if present
+                if prompt in decoded_output:
+                    decoded_output = decoded_output.replace(prompt, "").strip()
+                
+                # Capitalize the first letter of each word
+                decoded_output = decoded_output.title()
+                
                 # Display result
                 st.success("Analysis Complete!")
                 st.write("**Model Output:**", decoded_output)
 
+
+
             except Exception as e:
                 st.error(f"Error: {str(e)}")