BenkHel commited on
Commit
a9830ef
·
verified ·
1 Parent(s): 44d690a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py CHANGED
@@ -1,5 +1,43 @@
 
 
 
 
1
  import sys
2
  import os
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  import argparse
4
  import time
5
  import subprocess
 
1
+ # ===================================================================
2
+ # ===== START: Zwangsupgrade für die transformers-Bibliothek =====
3
+ # ===================================================================
4
+ import subprocess
5
  import sys
6
  import os
7
+
8
+ try:
9
+ # Versuche, die benötigte Klasse zu importieren und die Version zu prüfen
10
+ import transformers
11
+ print(f"--- Gefundene transformers-Version: {transformers.__version__} ---")
12
+ if transformers.__version__ < "4.37.2":
13
+ print("--- Version ist zu alt. Upgrade wird erzwungen. ---")
14
+ raise ImportError
15
+ from transformers import LlavaMistralForCausalLM
16
+ print("--- Transformers-Version ist ausreichend. ---")
17
+
18
+ except (ImportError, ModuleNotFoundError):
19
+ print("--- Transformers-Version unzureichend oder nicht gefunden. Führe Upgrade durch... ---")
20
+ # Führe das Upgrade mit pip durch
21
+ # Das '--no-cache-dir' ist wichtig, um sicherzustellen, dass nichts aus dem Cache geladen wird
22
+ subprocess.check_call([
23
+ sys.executable, "-m", "pip", "install", "--no-cache-dir", "--upgrade", "transformers==4.37.2"
24
+ ])
25
+ print("--- Upgrade abgeschlossen. Der Space wird neu gestartet, um die Änderungen zu laden. ---")
26
+
27
+ # Diese Zeile signalisiert Hugging Face, dass ein Neustart erforderlich ist.
28
+ # Es kann einen Moment dauern, bis der Space neu startet.
29
+ os.kill(os.getpid(), 9)
30
+
31
+ # ===================================================================
32
+ # ===== ENDE: Zwangsupgrade für die transformers-Bibliothek ======
33
+ # ===================================================================
34
+
35
+
36
+ # HIER BEGINNT IHR NORMALER CODE (unverändert lassen)
37
+ import cumo.serve.gradio_web_server as gws
38
+ from transformers import AutoProcessor, LlavaMistralForCausalLM
39
+ from transformers import TextIteratorStreamer
40
+ # ... und so weiter
41
  import argparse
42
  import time
43
  import subprocess