Update app.py

app.py

@@ -61,7 +61,11 @@ class ContentHandler(HuggingFaceEndpoint.ContentHandler):
     accepts = "application/json"
     len_prompt = 0
 
-    
+     def transform_input(self, prompt: str, model_kwargs: Dict) -> bytes:
+        self.len_prompt = len(prompt)
+        input_str = json.dumps({"inputs": prompt, "parameters": {"max_new_tokens": 100, "stop": ["Human:"], "do_sample": False, "repetition_penalty": 1.1}})
+        return input_str.encode('utf-8')
+         
     def transform_output(self, output: bytes) -> str:
         response_json = output.decode('utf-8')
         res = json.loads(response_json)