Update app.py

app.py

@@ -85,19 +85,19 @@ def load_model(model_type, selected_model):
 
         if model_type == "Full Fine-Tuned":
         
-            model = AutoModelForCausalLM.from_pretrained(
-                selected_model,
-                torch_dtype=torch.bfloat16,
-                device_map="auto",
-                token=HF_TOKEN  # optional if using private repo
-            )
+             model = AutoModelForMaskedLM.from_pretrained(
+                 selected_model,
+                 torch_dtype=torch.bfloat16,  # or float32 for compatibility
+                 token=HF_TOKEN
+            ).to("cuda" if torch.cuda.is_available() else "cpu")
 
             #model = AutoModelForCausalLM.from_pretrained(
                  #selected_model,
        #         torch_dtype=torch.bfloat16,
        #         device_map="auto",
        #         token=HF_TOKEN
-       #     )
+       #     
+        
         else:
             base_model = AutoModelForCausalLM.from_pretrained(
                 BASE_MODEL_NAME,