Update app.py

missing .ckpt extention

app.py

@@ -205,7 +205,7 @@ def extract_model_files(uploaded_files: List) -> List[Path]:
     temp_dir = Path(tempfile.mkdtemp())
     
     # Supported model file extensions
-    supported_extensions = {'.safetensors', '.pt', '.bin'}
+    supported_extensions = {'.safetensors', '.pt', '.bin', '.ckpt'}
     
     for file_info in uploaded_files:
         file_path = Path(file_info.name)
@@ -355,7 +355,7 @@ def process_files_async(uploaded_files: List) -> str:
         
         if total_files == 0:
             print("❌ No supported model files found")
-            return "No supported model files (.safetensors, .pt, .bin) found in uploaded content."
+            return "No supported model files (.safetensors, .pt, .bin, '.ckpt') found in uploaded content."
 
         update_progress(0, total_files, "Processing files...")
 
@@ -461,10 +461,10 @@ def create_interface():
         gr.Markdown("""
         # 🚁 DataDrones Bulk Uploader
         
-        Upload multiple model files (`.safetensors`, `.pt`, `.bin`) or zip archives containing model files to datadrones.com.
+        Upload multiple model files (`.safetensors`, `.pt`, `.bin`, `.ckpt`) or zip archives containing model files to datadrones.com.
         
         **Features:**
-        - Supports direct model file uploads (.safetensors, .pt, .bin) and zip archives
+        - Supports direct model file uploads (.safetensors, .pt, .bin, .ckpt) and zip archives
         - Automatic hash checking to avoid duplicates
         - Metadata retrieval from Civitai and other sources
         - Real-time progress tracking
@@ -474,9 +474,9 @@ def create_interface():
         with gr.Row():
             with gr.Column(scale=2):
                 file_input = gr.File(
-                    label="Select model files (.safetensors, .pt, .bin) or .zip archives",
+                    label="Select model files (.safetensors, .pt, .bin, .ckpt) or .zip archives",
                     file_count="multiple",
-                    file_types=[".safetensors", ".pt", ".bin", ".zip"]
+                    file_types=[".safetensors", ".pt", ".bin", ".zip", ".ckpt"]
                 )
                 
                 upload_btn = gr.Button("🚀 Start Upload", variant="primary", size="lg")
@@ -484,7 +484,7 @@ def create_interface():
             with gr.Column(scale=1):
                 gr.Markdown("""
                 ### Instructions:
-                1. Select multiple model files (`.safetensors`, `.pt`, `.bin`) directly, or
+                1. Select multiple model files (`.safetensors`, `.pt`, `.bin`, `.ckpt`) directly, or
                 2. Upload `.zip` archives containing model files
                 3. Click "Start Upload" to begin processing
                 4. Monitor progress in real-time below