Update app.py

app.py

@@ -380,7 +380,17 @@ def greedy_decode(audio_tensor, orig_len, model, device, max_len=MAX_TEXT_LEN):
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model_state = {"model": None}
 
-MODEL_PATH = os.getenv("MODEL_PATH", "entire_model.pth")
+import gdown
+
+MODEL_PATH = "entire_model.pth"
+
+if not os.path.exists(MODEL_PATH):
+    print("Downloading model from Google Drive...")
+    gdown.download(
+        "https://drive.google.com/uc?id=16i7ui-yYIWppaylOG2ofkpRlPQWeE4nZ",
+        MODEL_PATH,
+        quiet=False
+    )
 
 
 @asynccontextmanager