Update app.py

app.py

@@ -7,10 +7,12 @@ class GroupTherapyAgent:
     def __init__(self, client):
         self.client = client
         self.max_length = 64
-
+        
     def get_response(self, user_question):
-        response = self.client(text=user_question, max_length=self.max_length, num_beams=4)
-        return response[0]['generated_text']
+        # Make an API call to the model and get the response
+        response = self.client(inputs=user_question, parameters={"max_length": self.max_length, "num_beams": 4}, options={"wait_for_model": True})
+        return response['generated_text']
+
 
 class GroupTherapyApplication:
     def __init__(self, client):