Mistral

app.py

@@ -10,8 +10,18 @@ import sys
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-# Update the model name to Llama 3 8B
-model_name = "meta-llama/Meta-Llama-3-8B-Instruct"
+import gradio as gr
+import spaces
+import torch
+from transformers import AutoTokenizer, AutoModel
+import plotly.graph_objects as go
+from huggingface_hub import HfApi
+from huggingface_hub import hf_hub_download
+import os
+import sys
+
+# Update the model name to Mistral 7B
+model_name = "mistralai/Mistral-7B-v0.1"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = None