Create utils.py

utils.py

@@ -0,0 +1,22 @@
+import os
+import requests
+import torch
+import streamlit as st
+
+def get_device():
+    return torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+def download_checkpoint(url, filename):
+    if os.path.exists(filename):
+        st.info(f"Checkpoint già presente: {filename}")
+        return
+    st.warning(f"Scaricamento checkpoint SAM in corso... ({filename})")
+    resp = requests.get(url, stream=True)
+    total = int(resp.headers.get("content-length", 0))
+    dl = 0
+    bar = st.progress(0)
+    with open(filename, "wb") as f:
+        for chunk in resp.iter_content(1_048_576):
+            f.write(chunk); dl += len(chunk)
+            bar.progress(int(dl / total * 100))
+    st.success("Download completato!")