Update app.py

app.py

@@ -1,50 +1,16 @@
-# ===================================================================
-# ===== START: Zwangsupgrade für die transformers-Bibliothek =====
-# ===================================================================
+
 import subprocess
 import sys
 import os
 
-try:
-    # Versuche, die benötigte Klasse zu importieren und die Version zu prüfen
-    import transformers
-    print(f"--- Gefundene transformers-Version: {transformers.__version__} ---")
-    if transformers.__version__ < "4.37.2":
-        print("--- Version ist zu alt. Upgrade wird erzwungen. ---")
-        raise ImportError
-    from transformers import LlavaMistralForCausalLM
-    print("--- Transformers-Version ist ausreichend. ---")
-
-except (ImportError, ModuleNotFoundError):
-    print("--- Transformers-Version unzureichend oder nicht gefunden. Führe Upgrade durch... ---")
-    # Führe das Upgrade mit pip durch
-    # Das '--no-cache-dir' ist wichtig, um sicherzustellen, dass nichts aus dem Cache geladen wird
-    subprocess.check_call([
-        sys.executable, "-m", "pip", "install", "--no-cache-dir", "--upgrade", "transformers==4.37.2"
-    ])
-    print("--- Upgrade abgeschlossen. Der Space wird neu gestartet, um die Änderungen zu laden. ---")
-    
-    # Diese Zeile signalisiert Hugging Face, dass ein Neustart erforderlich ist.
-    # Es kann einen Moment dauern, bis der Space neu startet.
-    os.kill(os.getpid(), 9)
-
-# ===================================================================
-# ===== ENDE: Zwangsupgrade für die transformers-Bibliothek ======
-# ===================================================================
-
-
-# HIER BEGINNT IHR NORMALER CODE (unverändert lassen)
-import cumo.serve.gradio_web_server as gws
-from transformers import AutoProcessor, LlavaMistralForCausalLM
 from transformers import TextIteratorStreamer
-# ... und so weiter
 import argparse
 import time
 import subprocess
 import spaces
 import cumo.serve.gradio_web_server as gws
 
-from transformers import AutoProcessor, LlavaMistralForCausalLM
+from transformers import AutoProcessor
 
 import datetime
 import json
@@ -55,6 +21,7 @@ from PIL import Image
 
 from cumo.conversation import (default_conversation, conv_templates, SeparatorStyle)
 from cumo.constants import LOGDIR
+from cumo.model.language_model.llava_mistral import LlavaMistralForCausalLM
 from cumo.utils import (build_logger, server_error_msg, violates_moderation, moderation_msg)
 import hashlib