Update app.py

app.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer, PeftModel
+from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
+from peft import PeftModel
 
 class ModelInput(BaseModel):
     prompt: str