Update app.py

app.py

@@ -3,7 +3,7 @@ import torch  # <-- Add this import
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextGenerationPipeline
 
 # Load the model and tokenizer in full precision and ensure it's on CPU
-model = AutoModelForCausalLM.from_pretrained("PAIXAI/Astrid-1B").to(dtype=torch.float32).cpu()
+model = AutoModelForCausalLM.from_pretrained("PAIXAI/Astrid-1B")
 tokenizer = AutoTokenizer.from_pretrained("PAIXAI/Astrid-1B")
 
 # Initialize the pipeline with the model and tokenizer to run on CPU