Yasu777 commited on
Commit
25d0975
·
verified ·
1 Parent(s): 0b81deb

Update keywords_processor.py

Browse files
Files changed (1) hide show
  1. keywords_processor.py +7 -15
keywords_processor.py CHANGED
@@ -23,18 +23,10 @@ def ngram_generator(main_text, other_texts):
23
  output_text = save_keywords(keywords)
24
  return ", ".join(keywords) if keywords else "No keywords", output_text
25
 
26
- with gr.Blocks() as demo:
27
- gr.Markdown("### N-gram Generator and Saver")
28
- main_text_input = gr.Textbox(label="メインキーワード")
29
- other_texts_input = gr.Textbox(label="その他のキーワード(カンマ区切り)")
30
- output_keywords = gr.Textbox(label="Generated N-grams")
31
- output_message = gr.Textbox(label="Output Message")
32
- submit_button = gr.Button("Generate and Save N-grams")
33
-
34
- submit_button.click(
35
- fn=ngram_generator,
36
- inputs=[main_text_input, other_texts_input],
37
- outputs=[output_keywords, output_message]
38
- )
39
-
40
- if __name__ == "__main__":
 
23
  output_text = save_keywords(keywords)
24
  return ", ".join(keywords) if keywords else "No keywords", output_text
25
 
26
+ if __name__ == "__main__":
27
+ import sys
28
+ main_text = sys.argv[1] if len(sys.argv) > 1 else ""
29
+ other_texts = sys.argv[2] if len(sys.argv) > 2 else ""
30
+ keywords, output_text = ngram_generator(main_text, other_texts)
31
+ print(keywords)
32
+ print(output_text)