Update app.py

app.py

@@ -7,8 +7,8 @@ client = InferenceClient(
 )
 
 class GroupTherapyAgent:
-    def __init__(self, model, tokenizer):
-        self.model = model
+        def __init__(self, client, tokenizer):
+        self.client = client
         self.tokenizer = tokenizer
         self.max_length = 64
 
@@ -19,12 +19,12 @@ class GroupTherapyAgent:
         return response
 
     def generate_response(self, input_ids):
-        output = self.model.generate(input_ids, max_length=self.max_length, num_beams=4)
+        output = self.client.generate(input_ids, max_length=self.max_length, num_beams=4)
         return output
 
 class GroupTherapyApplication:
-    def __init__(self, model, tokenizer):
-        self.agents = [GroupTherapyAgent(model, tokenizer) for _ in range(4)]
+    def __init__(self, client, tokenizer):
+        self.agents = [GroupTherapyAgent(client, tokenizer) for _ in range(4)]
 
     def get_advice(self, user_question):
         advice = []
@@ -33,7 +33,7 @@ class GroupTherapyApplication:
             advice.append(response)
         return advice
 
-app = GroupTherapyApplication(model, tokenizer)
+app = GroupTherapyApplication(client, tokenizer)
 advice = app.get_advice("I feel anxious when I have to speak in front of a group of people.")
 print(f"Advice from Agents:\n{advice}")