Update app.py

app.py

@@ -30,7 +30,7 @@ from transformers.utils import logging
 from transformers import AutoTokenizer, AutoModelForCausalLM  # isort: skip
 
 logger = logging.get_logger(__name__)
-model_name_or_path="internlm/internlm2_5-1_8b-chat"
+model_name_or_path="cloneQ/my_personal_assistant"
 
 @dataclass
 class GenerationConfig:
@@ -186,7 +186,7 @@ def load_model():
     #     trust_remote_code=True).to(torch.bfloat16).cuda())
     model = AutoModelForCausalLM.from_pretrained(
         model_name_or_path,
-        trust_remote_code=True).to(torch.float32)
+        trust_remote_code=True).to("cpu")
     tokenizer = AutoTokenizer.from_pretrained(model_name_or_path,
                                               trust_remote_code=True)
     return model, tokenizer