Update app.py

app.py

@@ -2,8 +2,17 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 
 
-from transformers import AutoModel
-model = AutoModel.from_pretrained("rishu834763/java-explainer-lora", dtype="auto")
+from transformers import AutoModelForCausalLM
+from peft import PeftModel
+
+# This automatically loads the base model mentioned in adapter_config.json
+model = AutoModelForCausalLM.from_pretrained(
+    "rishu834763/java-explainer-lora",
+    device_map="auto",
+    torch_dtype="auto",
+    trust_remote_code=True  # only needed for some gated/gated models
+)
+model = PeftModel.from_pretrained(model, "rishu834763/java-explainer-lora")
 
 def respond(
     message,