update space

app.py

@@ -1,3 +1,4 @@
+
 import os
 import tensorflow as tf
 from keras.models import load_model
@@ -9,6 +10,7 @@ from pymongo import MongoClient
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from concurrent.futures import ThreadPoolExecutor
 import torch
+import random
 
 # Ensure the Hugging Face token is set
 tok = os.environ.get('HF_Token')
@@ -160,6 +162,8 @@ print("Loading Llama 3.2 model and tokenizer...")
 model_name = "meta-llama/Llama-3.2-1B"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 try:
+    # Use a random seed to avoid potential CUDA random_device error
+    torch.manual_seed(random.randint(0, 10000))
     model = AutoModelForCausalLM.from_pretrained(model_name).to('cuda' if torch.cuda.is_available() else 'cpu')
     print("Llama 3.2 model and tokenizer loaded successfully.")
 except Exception as e: