Test

app.py

@@ -1,11 +1,11 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
+import os
 
 client = InferenceClient(
     "mistralai/Mistral-7B-Instruct-v0.1"
 )
 
-
 def format_prompt(message, history):
   prompt = "<s>"
   for user_prompt, bot_response in history:
@@ -90,7 +90,7 @@ css = """
 """
 
 with gr.Blocks(css=css) as demo:
-    gr.HTML("<h1><center>Mistral 7B Instruct<h1><center>")
+    gr.HTML("<h1><center>Kyle Space<h1><center>")
     gr.HTML("<h3><center>In this demo, you can chat with <a href='https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.1'>Mistral-7B-Instruct</a> model. 💬<h3><center>")
     gr.HTML("<h3><center>Learn more about the model <a href='https://huggingface.co/docs/transformers/main/model_doc/mistral'>here</a>. 📚<h3><center>")
     gr.ChatInterface(