Upload InternVL2 implementation

Dockerfile

@@ -27,6 +27,11 @@ 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 .
 
@@ -53,6 +58,9 @@ 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 port 7860 available for the app
 EXPOSE 7860
 

app_internvl2.py

@@ -9,6 +9,12 @@ import warnings
 # 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)
+
 # Suppress specific warnings that might be caused by package version mismatches
 warnings.filterwarnings("ignore", message=".*The 'nopython' keyword.*")
 warnings.filterwarnings("ignore", message=".*Torch is not compiled with CUDA enabled.*")
@@ -179,6 +185,7 @@ def create_interface():
             outputs=output_text,
             fn=process_image,
             cache_examples=True,
+            examples_dir=cache_dir
         )
     
     return demo
@@ -188,5 +195,5 @@ if __name__ == "__main__":
     # Create the Gradio interface
     demo = create_interface()
     
-    # Launch the interface
-    demo.launch(share=False) 
+    # Launch the interface with explicit cache directory
+    demo.launch(share=False, cache_dir=cache_dir)