niyaa commited on
Commit
f2f8150
1 Parent(s): 78198ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -11
app.py CHANGED
@@ -5,18 +5,11 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
5
 
6
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
 
8
- today = datetime.datetime.now()
9
- next_year = today.year + 1
10
- jan_1 = datetime.date(next_year, 1, 1)
11
- dec_31 = datetime.date(next_year, 12, 31)
12
 
13
- d = st.date_input(
14
- "Select the date range",
15
- (jan_1, datetime.date(next_year, 1, 7)),
16
- jan_1,
17
- dec_31,
18
- format="MM.DD.YYYY",
19
- )
20
 
21
  tokenizer = AutoTokenizer.from_pretrained("nickmuchi/sec-bert-finetuned-finance-classification")
22
  model = AutoModelForSequenceClassification.from_pretrained("nickmuchi/sec-bert-finetuned-finance-classification")
 
5
 
6
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
 
8
+ import datetime
9
+ import streamlit as st
 
 
10
 
11
+ d = st.date_input("Chose the Date for which you are interested in", value=None)
12
+ st.write('You selected:', d)
 
 
 
 
 
13
 
14
  tokenizer = AutoTokenizer.from_pretrained("nickmuchi/sec-bert-finetuned-finance-classification")
15
  model = AutoModelForSequenceClassification.from_pretrained("nickmuchi/sec-bert-finetuned-finance-classification")