Update app.py

app.py

@@ -40,6 +40,11 @@ class GradioLLM(Runnable):
         super().__init__()  # Call the constructor of Runnable if necessary
         self.client = Client(client_url)
 
+    def invoke(self, prompt, **kwargs):
+        # Implement the invoke method here
+        result = self.client.predict(prompt, prompt, api_name="/predict")
+        return result['data'][0]  # Adjust based on the actual response format    
+
     def generate(self, prompt, **kwargs):
         # Assuming the API expects 'prompt' and 'system_prompt' as inputs
         result = self.client.predict(prompt, prompt, api_name="/predict")