Update app.py

app.py

@@ -203,17 +203,60 @@
 # interface.launch(debug=True)
 #!/usr/bin/env python
 
+# #!/usr/bin/env python
+
+# import os
+# import torch
+# from multimodal_app import create_interface
+
+# # Optional: Set torch threads to limit CPU usage
+# torch.set_num_threads(4)
+
+# # Create and launch the interface
+# demo = create_interface()
+# demo.queue()  # Enable queuing for better handling of multiple requests
+# demo.launch()
 #!/usr/bin/env python
 
 import os
 import torch
-from multimodal_app import create_interface
+import sys
+import argparse
 
-# Optional: Set torch threads to limit CPU usage
+# Set environment variables for better compatibility
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
+
+# Set torch threads to limit CPU usage
 torch.set_num_threads(4)
 
-# Create and launch the interface
-demo = create_interface()
-demo.queue()  # Enable queuing for better handling of multiple requests
-demo.launch()
+# Parse command line arguments
+parser = argparse.ArgumentParser(description="Multimodal Image Description App")
+parser.add_argument("--peft-model", type=str, default="/content/drive/MyDrive/V6_Checkpoints/Epoch_peft_1",
+                   help="Path to PEFT model")
+parser.add_argument("--port", type=int, default=7860,
+                   help="Port to run the server on")
+args = parser.parse_args()
 
+try:
+    from multimodal_app import create_interface, load_model
+    
+    # Preload the model with PEFT path
+    print(f"Preloading model with PEFT path: {args.peft_model}")
+    load_model(args.peft_model)
+    
+    # Create and launch the interface
+    demo = create_interface()
+    
+    # Launch with proper settings for stability
+    demo.launch(
+        share=False,              # Set to True if you want a public link
+        debug=True,               # Enable debug for better error messages
+        server_name="0.0.0.0",    # Listen on all interfaces  
+        server_port=args.port,    # Port from arguments
+        show_api=False            # Hide API docs for simplicity
+    )
+except Exception as e:
+    print(f"Error starting application: {e}")
+    import traceback
+    traceback.print_exc()
+    sys.exit(1)