Update app.py

app.py

@@ -72,8 +72,7 @@ def get_model_and_tokenizer():
         tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
         model = AutoModelForCausalLM.from_pretrained(
             MODEL_NAME,
-            torch_dtype=torch.float16,
-            device_map="auto"
+            torch_dtype=torch.float16
         )
         logger.info("Model and tokenizer loaded successfully")
         return model, tokenizer
@@ -1493,8 +1492,7 @@ class EducationalChatbot:
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
         self.model = AutoModelForCausalLM.from_pretrained(
             self.model_name,
-            torch_dtype=torch.float16,
-            device_map="auto"
+            torch_dtype=torch.float16
         )
         logger.info("Educational chatbot model loaded")