Update app.py

app.py

@@ -1,5 +1,30 @@
 import gradio as gr
 # from humanize import paraphrase_text
+from gradio_client import Client
+from ai_generate import generate
+
+client = Client("polygraf-ai/Humanizer")
+def humanize(
+    text, 
+    model, 
+    temperature=1.2, 
+    repetition_penalty=1,
+    top_k=50,
+    length_penalty=1,
+)
+    ai_text = generate(f"Write an article about the topic: {text}")
+    print(f"AI Generated: {ai_text}")
+    ai_text = ai_text["choices"][0]["message"]["content"]
+    result = client.predict(
+    		text=text,
+    		model_name=model,
+    		temperature=temperature,
+    		repetition_penalty=repetition_penalty,
+    		top_k=top_k,
+    		length_penalty=length_penalty,
+    		api_name="/paraphrase_text"
+    )
+    return result
 
 with gr.Blocks() as demo:
     gr.Markdown("# Polygraf Writer")
@@ -52,7 +77,7 @@ with gr.Blocks() as demo:
                 gr.Markdown("Penalizes shorter outputs.")
     
         process_button.click(
-            fn=paraphrase_text,
+            fn=humanize,
             inputs=[
                 input_topic,
                 model_dropdown,