politweet / app.py
Demea9000's picture
made gradio runnable
15ca093
raw history blame
No virus
671 Bytes
import gradio as gr
from textclassifier import TextClassifier as tc
def main(from_date, to_date, user_name):
"""
Main function. Runs the program.
:return: None
"""
from_date = from_date
to_date = to_date
user_name = user_name
text_classifier = tc.TextClassifier(from_date=from_date, to_date=to_date, user_name=user_name, num_tweets=20)
text_classifier.run_main_pipeline()
return text_classifier.get_dataframe()
def greet(name):
return "Hello " + name + "!"
if __name__ == "__main__":
demo = gr.Interface(
fn=main,
inputs=['text', 'text', 'text'],
outputs="dataframe",
)
demo.launch()