updated streamlit file

.ipynb_checkpoints/Dockerfile-checkpoint

@@ -14,6 +14,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 RUN pip install --no-cache-dir torch
 
+RUN pip install --no-cache-dir --upgrade transformers
+
 # Install supervisor to manage multiple processes
 RUN apt-get update && apt-get install -y supervisor \
     && apt-get clean && rm -rf /var/lib/apt/lists/*

.ipynb_checkpoints/streamlit_app-checkpoint.py

@@ -16,6 +16,7 @@ def load_model():
     return model, processor
 
 model, processor = load_model()
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 st.title("Image Captioning with Fine-Tuned BLiPv2 Model")
 

Dockerfile

@@ -14,6 +14,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 RUN pip install --no-cache-dir torch
 
+RUN pip install --no-cache-dir --upgrade transformers
+
 # Install supervisor to manage multiple processes
 RUN apt-get update && apt-get install -y supervisor \
     && apt-get clean && rm -rf /var/lib/apt/lists/*

streamlit_app.py

@@ -16,6 +16,7 @@ def load_model():
     return model, processor
 
 model, processor = load_model()
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 st.title("Image Captioning with Fine-Tuned BLiPv2 Model")