updates

app.py

@@ -14,6 +14,8 @@ import io
 from utils.model_utils import get_model_caption
 from utils.image_utils import overlay_caption
 
+!pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
+
 @st.cache_resource
 def load_models():
     base_model = AutoModelForCausalLM.from_pretrained("google/gemma-2b")