Update app.py

app.py

@@ -1,25 +1,34 @@
-"""pip install gradio
-pip install openai
-pip install PyPDF2
-pip install tiktoken
-pip install python-pptx
-apt-get install git"""
-import gradio as gr
-
-git clone https://github.com/facebookresearch/segment-anything-2
 %cd /content/segment-anything-2
 
-git checkout sam2.1
-# Install the package
-pip install .
+import os
+import urllib.request
+
+model_urls = {
+    "sam2_hiera_tiny.pt": "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_tiny.pt",
+    "sam2_hiera_small.pt": "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_small.pt",
+    "sam2_hiera_base_plus.pt": "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_base_plus.pt",
+    "sam2_hiera_large.pt": "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_large.pt",
+}
+
+def download_models():
+    for filename, url in model_urls.items():
+        if not os.path.exists(filename):
+            print(f"Downloading {filename}...")
+            urllib.request.urlretrieve(url, filename)
+        else:
+            print(f"{filename} already exists, skipping download.")
 
-wget -O sam2_hiera_tiny.pt "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_tiny.pt"
+# Call it at startup
+download_models()
+
+"""wget -O sam2_hiera_tiny.pt "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_tiny.pt"
 wget -O sam2_hiera_small.pt "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_small.pt"
 wget -O sam2_hiera_base_plus.pt "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_base_plus.pt"
 wget -O sam2_hiera_large.pt "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_large.pt"
+"""
 
-import os
-#import gradio as gr
+#import os
+import gradio as gr
 import numpy as np
 import pandas as pd
 import cv2
@@ -45,12 +54,11 @@ from time import sleep
 from csv import writer
 from tqdm import tqdm
 from scipy import spatial
-from pptx import Presentation  # for PowerPoint
+from pptx import Presentation
 from PyPDF2 import PdfReader
 from openai import OpenAI
 from IPython.display import display, Markdown, Latex, HTML
 from transformers import GPT2Tokenizer
-#from transformers import AutoTokenizer
 from google.colab import files  # for uploading files
 from termcolor import colored  # for colored text output