Update app.py

app.py

@@ -1,9 +1,10 @@
 
 import streamlit as st
-from transformers import AutoModelForCausalLM, AutoTokenizer
-model_name = "mistralai/Mixtral-8x7B-Instruct-v0.1"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+from huggingface_hub import InferenceClient
+
+client = InferenceClient(
+    "mistralai/Mistral-7B-Instruct-v0.1"
+)
 
 class GroupTherapyAgent:
     def __init__(self, model, tokenizer):