Update app.py

app.py

@@ -1,12 +1,9 @@
-from transformers import AutoTokenizer, MistralForCausalLM
-import torch
-import gradio as gr
-import random
-from textwrap import wrap
 from transformers import AutoConfig, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, MistralForCausalLM
 from peft import PeftModel, PeftConfig
 import torch
 import gradio as gr
+import random
+from textwrap import wrap
 
 # Functions to Wrap the Prompt Correctly
 def wrap_text(text, width=90):
@@ -103,8 +100,8 @@ class ChatBot:
 bot = ChatBot()
 
 title = "👋🏻Welcome to Tonic's MistralMed Chat🚀"
-description = "You can use this Space to test out the current model (MistralMed) or duplicate this Space and use it for any other model on 🤗HuggingFace. Join me on Discord to build together."
-examples = [["What is the proper treatment for buccal herpes?", "Please provide information on the most effective antiviral medications and home remedies for treating buccal herpes."]]
+description = "You can use this Space to test out the current model [(Tonic/MistralMed)](https://huggingface.co/Tonic/MistralMed) or duplicate this Space and use it locally or on 🤗HuggingFace. [Join me on Discord to build together](https://discord.gg/VqTxc76K3u)."
+examples = [["[Question:] What is the proper treatment for buccal herpes?", "You are a medicine and public health expert, you will recieve a question, answer the question and complete answer"]]
 
 iface = gr.Interface(
     fn=bot.predict,