Update app.py

app.py

@@ -9,6 +9,13 @@ demo = gr.ChatInterface(mental_chat)
 demo.launch()
 """
 
+import gradio as gr
+import torch
+from peft import PeftModel, PeftConfig
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+# ##### ##### ##### ##### #####
+
 peft_model_id = "charansr/llama2-7b-chat-hf-therapist"
 config = PeftConfig.from_pretrained(peft_model_id)
 newmodel = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, return_dict=True, load_in_8bit=True, device_map='auto')