Update app.py

app.py

@@ -1,7 +1,7 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
-
-
+import os
+client = InferenceClient("Satyam-Singh/LLaVa-Large-Language-Virtual-Assistant")
 
 def format_prompt(message, history):
   prompt = "<s>"
@@ -10,7 +10,8 @@ def format_prompt(message, history):
     prompt += f" {bot_response}</s> "
   prompt += f"[INST] {message} [/INST]"
   return prompt
-client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
+    
+client = InferenceClient(os.environ.get('LLAVA'))
 
 def generate(
     prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,