Update app2.py

app2.py

@@ -1,3 +1,15 @@
+import os
+import shutil
+
+# --- SEARCH AND DESTROY POISONED CACHE ---
+corrupted_dir = "/root/.cache/huggingface/hub/models--google--umt5-base"
+if os.path.exists(corrupted_dir):
+    print("[SYSTEM] Found corrupted UMT5 cache. Deleting...")
+    shutil.rmtree(corrupted_dir, ignore_errors=True)
+else:
+    print("[SYSTEM] Cache is clean.")
+
+# --- YOUR ORIGINAL CODE STARTS HERE ---
 from flask import Flask, request, jsonify
 from flask_sock import Sock
 from transformers import AutoModel
@@ -19,12 +31,11 @@ model = AutoModel.from_pretrained(
     trust_remote_code=True
 )
 
-# 2. M1 Architecture Override
+# 2. Cloud Architecture Override
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = model.to(device)
 print(f"[SYSTEM] Model loaded successfully onto device: {device}")
 
-
 # --- THE NEW WEBSOCKET PIPELINE ---
 @sock.route('/api/generate_stream')
 def stream_motion(ws):
@@ -65,5 +76,5 @@ def stream_motion(ws):
             break
 
 if __name__ == '__main__':
-    # --- CHANGE THE PORT TO 7860 ---
+    # --- PORT 7860 FOR HUGGING FACE ---
     app.run(host='0.0.0.0', port=7860, debug=False)