Update app.py

app.py

@@ -1,30 +1,30 @@
 import gradio as gr
-# from humanize import paraphrase_text
+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}")
-    ai_text = ai_text.choices[0].message.content
-    print(f"AI Generated: {ai_text}")
-    result = client.predict(
-    		text=ai_text,
-    		model_name=model,
-    		temperature=temperature,
-    		repetition_penalty=repetition_penalty,
-    		top_k=top_k,
-    		length_penalty=length_penalty,
-    		api_name="/paraphrase_text"
-    )
-    return ai_text, result
+# 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}")
+#     ai_text = ai_text.choices[0].message.content
+#     print(f"AI Generated: {ai_text}")
+#     result = client.predict(
+#     		text=ai_text,
+#     		model_name=model,
+#     		temperature=temperature,
+#     		repetition_penalty=repetition_penalty,
+#     		top_k=top_k,
+#     		length_penalty=length_penalty,
+#     		api_name="/paraphrase_text"
+#     )
+#     return ai_text, result
 
 with gr.Blocks() as demo:
     gr.Markdown("# Polygraf Writer")
@@ -79,7 +79,7 @@ with gr.Blocks() as demo:
                 gr.Markdown("Penalizes shorter outputs.")
     
         process_button.click(
-            fn=humanize,
+            fn=paraphrase_text,
             inputs=[
                 input_topic,
                 model_dropdown,