Upload InternVL2 implementation

Dockerfile

@@ -3,14 +3,15 @@ FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
-ENV HF_HOME=/root/.cache/huggingface
-ENV TRANSFORMERS_CACHE=/root/.cache/huggingface/transformers
+ENV HF_HOME=/app/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/transformers
 ENV MPLCONFIGDIR=/tmp/matplotlib
 
 # Create necessary directories with proper permissions
-RUN mkdir -p /root/.cache/huggingface/transformers && \
+RUN mkdir -p /app/.cache/huggingface/transformers && \
     mkdir -p /tmp/matplotlib && \
-    chmod -R 777 /root/.cache && \
+    mkdir -p /app/gradio_cached_examples && \
+    chmod -R 777 /app && \
     chmod -R 777 /tmp/matplotlib
 
 # Install system dependencies
@@ -27,11 +28,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 # Create a working directory
 WORKDIR /app
 
-# Create and set permissions for Gradio cache directory
-RUN mkdir -p /app/gradio_cached_examples && \
-    chmod 777 /app/gradio_cached_examples && \
-    chmod 777 /app
-
 # Copy requirements file
 COPY requirements.txt .
 
@@ -58,8 +54,11 @@ RUN pip3 install --no-cache-dir --upgrade pip && \
 # Copy the application files
 COPY . .
 
-# Set permissions for the entire app directory
-RUN chmod -R 777 /app
+# Make sure the runtime directories exist and have proper permissions
+RUN mkdir -p gradio_cached_examples && \
+    chmod -R 777 gradio_cached_examples && \
+    mkdir -p .cache/huggingface/transformers && \
+    chmod -R 777 .cache
 
 # Make port 7860 available for the app
 EXPOSE 7860

app_internvl2.py

@@ -5,15 +5,36 @@ import time
 import numpy as np
 import torch
 import warnings
+import stat
 
 # Set environment variables
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"
 
-# Ensure cache directories exist with proper permissions
-cache_dir = os.path.join(os.getcwd(), "gradio_cached_examples")
-os.makedirs(cache_dir, exist_ok=True)
-if os.path.exists(cache_dir):
-    os.chmod(cache_dir, 0o777)
+# Ensure all cache directories exist with proper permissions
+def setup_cache_directories():
+    # Gradio cache directory
+    cache_dir = os.path.join(os.getcwd(), "gradio_cached_examples")
+    os.makedirs(cache_dir, exist_ok=True)
+    
+    # HuggingFace cache directories
+    hf_cache = os.path.join(os.getcwd(), ".cache", "huggingface")
+    transformers_cache = os.path.join(hf_cache, "transformers")
+    os.makedirs(hf_cache, exist_ok=True)
+    os.makedirs(transformers_cache, exist_ok=True)
+    
+    # Set permissions
+    try:
+        for directory in [cache_dir, hf_cache, transformers_cache]:
+            if os.path.exists(directory):
+                os.chmod(directory, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)  # 0o777
+                print(f"Set permissions for {directory}")
+    except Exception as e:
+        print(f"Warning: Could not set permissions: {str(e)}")
+    
+    return cache_dir
+
+# Set up cache directories
+cache_dir = setup_cache_directories()
 
 # Suppress specific warnings that might be caused by package version mismatches
 warnings.filterwarnings("ignore", message=".*The 'nopython' keyword.*")
@@ -175,18 +196,21 @@ def create_interface():
         """)
         
         # Examples
-        gr.Examples(
-            examples=[
-                ["data_temp/page_2.png", "general"],
-                ["data_temp/page_2.png", "text"],
-                ["data_temp/page_2.png", "chart"]
-            ],
-            inputs=[input_image, analysis_type],
-            outputs=output_text,
-            fn=process_image,
-            cache_examples=True,
-            examples_dir=cache_dir
-        )
+        try:
+            gr.Examples(
+                examples=[
+                    ["data_temp/page_2.png", "general"],
+                    ["data_temp/page_2.png", "text"],
+                    ["data_temp/page_2.png", "chart"]
+                ],
+                inputs=[input_image, analysis_type],
+                outputs=output_text,
+                fn=process_image,
+                cache_examples=True,
+                examples_dir=cache_dir
+            )
+        except Exception as e:
+            print(f"Warning: Could not load examples: {str(e)}")
     
     return demo
 
@@ -195,5 +219,5 @@ if __name__ == "__main__":
     # Create the Gradio interface
     demo = create_interface()
     
-    # Launch the interface with explicit cache directory
-    demo.launch(share=False, cache_dir=cache_dir) 
+    # Launch the interface with explicit cache directory and without authentication
+    demo.launch(share=False, cache_dir=cache_dir, server_name="0.0.0.0")