Upload app.py

app.py

@@ -1,3 +1,17 @@
+from transformers import pipeline, AutoTokenizer
+import pandas as pd
+import numpy as np
+import torch
+import streamlit as st
+
+USE_GPU = True
+
+if USE_GPU and torch.cuda.is_available():
+    device = torch.device("cuda:0")
+else: 
+    device = torch.device('cpu')
+
+
 WORD_PROBABILITY_THRESHOLD = 0.02
 #WORD_PROBABILITY_THRESHOLD_ENGLISH = 0.02
 #WORD_PROBABILITY_THRESHOLD_CHINESE = 0.02