Update app.py

app.py

@@ -7,6 +7,10 @@ import io
 import torch
 import clip
 import tensorflow as tf
+import os
+
+
+
 
 app = FastAPI()
 
@@ -14,8 +18,14 @@ app = FastAPI()
 tflite_model = tf.lite.Interpreter(model_path="resnet_model.tflite")
 tflite_model.allocate_tensors()
 
+# clip_device = "cuda" if torch.cuda.is_available() else "cpu"
+# clip_model, clip_preprocess = clip.load("ViT-B/32", device=clip_device)
 clip_device = "cuda" if torch.cuda.is_available() else "cpu"
-clip_model, clip_preprocess = clip.load("ViT-B/32", device=clip_device)
+
+# Create a writable models directory
+os.makedirs("models", exist_ok=True)
+
+clip_model, clip_preprocess = clip.load("ViT-B/32", device=clip_device, download_root="./models")
 
 
 # Class names