sentence_fix / app.py
taufiqdp's picture
Update app.py
41dbe2e
raw
history blame
1.72 kB
import gradio as gr
import google.generativeai as palm
import os
# Get your own API key on https://developers.generativeai.google/products/palm
api_key = os.environ["api_key"]
palm.configure(api_key=api_key)
defaults = {
'model': 'models/text-bison-001',
'temperature': 0.7,
'candidate_count': 1,
'top_k': 40,
'top_p': 0.95,
'max_output_tokens': 1024,
'stop_sequences': [],
'safety_settings': [
{"category": "HARM_CATEGORY_DEROGATORY", "threshold": 3},
{"category": "HARM_CATEGORY_TOXICITY", "threshold": 3},
{"category": "HARM_CATEGORY_VIOLENCE", "threshold": 3},
{"category": "HARM_CATEGORY_SEXUAL", "threshold": 3},
{"category": "HARM_CATEGORY_MEDICAL", "threshold": 3},
{"category": "HARM_CATEGORY_DANGEROUS", "threshold": 3}]
}
with open("example.txt") as f:
example = f.read()
with gr.Blocks() as app:
def chat(text):
response = palm.generate_text(
**defaults,
prompt = f"""Rewrite the following sentence to fix the grammar issues and correct the sentence.
{example}
input: {text}
fixed"""
)
return response.result
with gr.Column():
text = gr.Textbox(lines=4, label="Text", max_lines=4, placeholder="Write something awesome. It will be corrected automatically.")
with gr.Column():
output = gr.Textbox(lines=4, label="Output", max_lines=4, show_copy_button=True)
with gr.Row():
clr_btn = gr.ClearButton([text, output], variant="primary")
btn = gr.Button("Submit")
btn.click(fn=chat, inputs=text, outputs=output)
if __name__ == "__main__":
app.launch()