Update app.py

app.py

@@ -5,11 +5,14 @@ from transformers import AutoModel, AutoConfig
 
 hf_token = os.environ.get("HF_TOKEN")
 
-local_config_path = "HusseinEid/lora_model/tree/main/adapter_config.json"
+model_repo = "HusseinEid/lora_model"
+config_file = "adapter_config.json"
 
 try:
-    config = AutoConfig.from_pretrained(local_config_path)
-    model = AutoModel.from_pretrained("HusseinEid/lora_model", config=config, token=hf_token)
+    # Load the configuration directly from the repository
+    config = AutoConfig.from_pretrained(model_repo, config=config_file, token=hf_token)
+    # Load the model from Hugging Face using the config
+    model = AutoModel.from_pretrained(model_repo, config=config, token=hf_token)
 except OSError as e:
     print(f"Error: {e}")