Update space

app.py

@@ -69,7 +69,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import os
 
 # Define model names
-MODEL_1_PATH = "./adapter_model.safetensors"  # Local path inside Space
+# MODEL_1_PATH = "./adapter_model.safetensors"  # Local path inside Space
 ###
 MODEL_1_PATH = "Priyanka6/fine-tuning-inference"
 ###
@@ -110,9 +110,10 @@ TOKENIZER_NAME = "sarvamai/sarvam-1"
 tokenizer = AutoTokenizer.from_pretrained(TOKENIZER_NAME)
 
 # Function to load a model
-def load_model(model_choice,trimmed_adapter_path):
+def load_model(model_choice):
     if model_choice == "Hugging face dataset":
         model = AutoModelForCausalLM.from_pretrained("./", torch_dtype=torch.float16, device_map="auto")
+        trimmed_adapter_path = os.path.join("Priyanka6/fine-tuning-inference", "adapter_model_trimmed.safetensors")
         model.load_adapter(trimmed_adapter_path, "safe_tensors")  # Load safetensors adapter
     else:
         model = AutoModelForCausalLM.from_pretrained(MODEL_2_NAME)