Update app.py

app.py

@@ -1,7 +1,12 @@
 import gradio as gr
+from peft import PeftModel
+from transformers import AutoModelForCausalLM
 
 def greet(name):
     return "Hello " + name + "!!"
 
+base_model = AutoModelForCausalLM.from_pretrained("davidkim205/komt-mistral-7b-v1")
+model = PeftModel.from_pretrained(base_model, "interview")
+
 demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo.launch()