Update app.py

app.py

@@ -13,10 +13,17 @@ os.environ["HF_HOME"] = "/tmp"
 os.environ["TRANSFORMERS_CACHE"] = "/tmp"
 
 
-model_id = "rabiyulfahim/Python_coding_model"
+# ✅ Base + Adapter
+from peft import PeftModel
+
+base_model_id = "microsoft/DialoGPT-small"
+adapter_id = "rabiyulfahim/Python_coding_model"
+
+tokenizer = AutoTokenizer.from_pretrained(base_model_id, cache_dir="/tmp")
+base_model = AutoModelForCausalLM.from_pretrained(base_model_id, cache_dir="/tmp")
+model = PeftModel.from_pretrained(base_model, adapter_id, cache_dir="/tmp")
+
 
-tokenizer = AutoTokenizer.from_pretrained(model_id, cache_dir="/tmp")
-model = AutoModelForCausalLM.from_pretrained(model_id, cache_dir="/tmp")
 
 
 app = FastAPI(title="QA GPT2 API UI", description="Serving HuggingFace model with FastAPI")