Update app.py

app.py

@@ -1,6 +1,9 @@
-import gradio as gr
-import torch
-from transformers import AutoTokenizer, AutoModel, BertConfig  # Use BertConfig for your Mistral model
+from transformers import AutoTokenizer, AutoModelForCausalLM  
+import gradio as gr  
+  
+# api token for huggingface.co  
+api_token = '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'  
+  
 
 # Use the base model's ID
 base_model_id = "mistralai/Mistral-7B-v0.1"
@@ -30,19 +33,19 @@ class ChatBot:
         return response
 
 bot = ChatBot()
-
-title = "👋🏻Welcome to Tonic's EZ 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](https://discord.gg/fpEPNZGsbt) to build together."
-examples = [["What is the boiling point of nitrogen"]]
-
-iface = gr.Interface(
-    fn=bot.predict,
-    title=title,
-    description=description,
-    examples=examples,
-    inputs="text",
-    outputs="text",
-    theme="ParityError/Anime"
-)
-
-iface.launch()
+  
+title = "👋🏻Welcome to Tonic's EZ 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](https://discord.gg/fpEPNZGsbt) to build together."  
+examples = [["What is the boiling point of nitrogen"]]  
+  
+iface = gr.Interface(  
+    fn=bot.predict,  
+    title=title,  
+    description=description,  
+    examples=examples,  
+    inputs="text",  
+    outputs="text",  
+    theme="ParityError/Anime"  
+)  
+  
+iface.launch()