Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import Blip2Processor
 from peft import PeftModel
 import streamlit as st
 from PIL import Image
-import torch
+#import torch
 import os
 
 preprocess_ckp = "Salesforce/blip2-opt-2.7b" #Checkpoint path used for perprocess image
@@ -42,12 +42,13 @@ def main():
     #init_model()
 
     #Select few sample images for the catagory of cloths
-    option = st.selectbox('Sample images ?', ('cap', 'tee', 'dress'))
-    file_name = st.file_uploader("Upload image")
+    option = st.selectbox('Select from sample an images', ('cap', 'tee', 'dress'), index = -1)
+    st.text("OR")
+    file_name = st.file_uploader("Upload an image")
 
     if file_name is None and option is not None:
 
-        file_name = os.join.path(sample_img_path, option)
+        file_name = os.path.join(sample_img_path, option)
 
     if file_name is not None: