Update main.py

main.py

@@ -49,6 +49,7 @@ def response_generator(query):
     endpoint_url = ("https://api-inference.huggingface.co/models/"+ model)
     model_kwargs = {"temperature" : temperature,
                     "max_new_tokens" : max_tokens,
+                    "repetition_penalty" : 1.1,
                     "return_full_text" : False}
     hf = HuggingFaceEndpoint(
         endpoint_url=endpoint_url,