Update app.py

app.py

@@ -1,7 +1,14 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
 import streamlit as st
+
+from peft import PeftModel, PeftConfig
+
 tokenizer = AutoTokenizer.from_pretrained("SSahas/openai_community_med_e3")
-model = AutoModelForCausalLM.from_pretrained("SSahas/openai_community_med_e3")
+#model = AutoModelForCausalLM.from_pretrained("SSahas/openai_community_med_e3")
+
+config = PeftConfig.from_pretrained("SSahas/openai_community_med_e3")
+model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2-medium")
+model = PeftModel.from_pretrained(model, "SSahas/openai_community_med_e3")
 
 
 def response_generator(prompt):