Update space

app.py

@@ -204,9 +204,11 @@ trimmed_adapter_path = trim_adapter_weights(MODEL_1_PATH)
 tokenizer = AutoTokenizer.from_pretrained(TOKENIZER_NAME)
 
 # Load the model
-model = AutoModelForCausalLM.from_pretrained(
-    MODEL_1_PATH, torch_dtype=torch.float16, device_map="auto"
-)
+# model = AutoModelForCausalLM.from_pretrained(
+#     MODEL_1_PATH, torch_dtype=torch.float16, device_map="auto"
+# )
+
+model = AutoModelForCausalLM.from_pretrained("Priyanka6/fine-tuning-inference", use_auth_token=True)
 
 # Load the trimmed adapter
 model.load_adapter(trimmed_adapter_path, "safe_tensors")