Update app.py

app.py

@@ -4,27 +4,35 @@ 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
+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"
+    # )
+    result = paraphrase_text(
+    		text=ai_text,
+    		model_name=model,
+    		temperature=temperature,
+    		repetition_penalty=repetition_penalty,
+    		top_k=top_k,
+    		length_penalty=length_penalty,
+    )
+    return ai_text, result
 
 with gr.Blocks() as demo:
     gr.Markdown("# Polygraf Writer")
@@ -79,7 +87,7 @@ with gr.Blocks() as demo:
                 gr.Markdown("Penalizes shorter outputs.")
     
         process_button.click(
-            fn=paraphrase_text,
+            fn=humanize,
             inputs=[
                 input_topic,
                 model_dropdown,