Update app.py

app.py

@@ -10,7 +10,7 @@ import subprocess
 import spaces
 import cumo.serve.gradio_web_server as gws
 
-from transformers import AutoProcessor
+from transformers import AutoProcessor,AutoTokenizer, AutoImageProcessor
 
 import datetime
 import json
@@ -64,8 +64,11 @@ model = LlavaMistralForCausalLM.from_pretrained(
 )
 
 # Weisen Sie die Komponenten den alten Variablennamen zu, damit der restliche Code funktioniert
-tokenizer = processor.tokenizer
-image_processor = processor.image_processor
+tokenizer = AutoTokenizer.from_pretrained(model_path)
+try:
+    image_processor = AutoImageProcessor.from_pretrained(model_path)
+except Exception:
+    image_processor = None
 
 # Setzen Sie die Kontextlänge (falls der restliche Code sie benötigt)
 context_len = model.config.max_position_embeddings