add upload function

app.py

@@ -3,6 +3,7 @@ from transformers import AutoModelForSequenceClassification
 from transformers import AutoTokenizer
 import gradio as gr
 import torch
+import pandas as pd
 
 # Load trained model
 model = AutoModelForSequenceClassification.from_pretrained("chkla/parlbert-topic-german")
@@ -43,11 +44,17 @@ with gr.Blocks() as demo:
     )
 
     # Upload file
-    # file_output = gr.File()
-    # upload_button = gr.UploadButton("Click to Upload a File", type="file", accept=".csv", label="File", outputs=[file_output])
+    file_output = gr.File()
+    upload_button = gr.UploadButton("Click to Upload a File", type="file", accept=".csv", label="File", outputs=[file_output])
+    gr.Dataframe(pd.DataFrame(file_output))
+
 
     # Link to paper and Github repo
     gr.Markdown('''For more details: You can read our [paper](http://www.lrec-conf.org/proceedings/lrec2022/workshops/ParlaCLARINIII/pdf/2022.parlaclariniii-1.13.pdf) or access our [code](https://github.com/chkla/FrameASt).''')
     gr.Markdown('''Enjoy and stay tuned 🚀''')
 
-demo.launch()
+def main():
+    demo.launch()
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -3,4 +3,5 @@ hatesonar==0.0.7
 torch==1.8.0
 transformers==4.10.0
 scikit_learn==0.23.1
-textblob==0.15.0
+textblob==0.15.0
+pandas