dfedukov commited on
Commit
af5e809
·
verified ·
1 Parent(s): 97377d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +31 -7
app.py CHANGED
@@ -11,14 +11,38 @@ ind_to_target = {ind: target for target, ind in target_to_ind.items()}
11
 
12
 
13
  st.title('papers_classifier 🤓')
14
- st.text("Hey! I'm papers_classifier and I'm here to help you with answering the question 'WTF is this paper about?\n'
15
- According to arXiv there are 8 different fields of study - Computer Science, Economics, Electrical Engineering and Systems Science, Mathematics, Physics, Quantitative Biology, \
16
- Quantitative Finance and Statistics. So, everything I'll tell you will be about these eight gentlemen.\n
17
- You need to give me paper's title and (if you have one) it's abstract. Also you need to choose classification mode - there are 2 of them:
18
- best prediction and top 95% which means that you'll see as many classes as I need to show you to be confident with probability at least 0.95 that the correct one is among them.\n
19
- After that you need to press the Get prediction button and I'll tell you to which fields of study this paper is related. \n
20
- ")
21
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
  @st.cache_resource
23
  def load_model_and_tokenizer():
24
  model_name = 'distilbert/distilbert-base-cased'
 
11
 
12
 
13
  st.title('papers_classifier 🤓')
 
 
 
 
 
 
 
14
 
15
+ @st.cache_data
16
+ def display_intro():
17
+ intro_text = """
18
+ Hey! I'm papers_classifier and I'm here to help you with answering the question 'WTF is this paper about?'
19
+
20
+ According to arXiv there are 8 different fields of study:
21
+ - Computer Science
22
+ - Economics
23
+ - Electrical Engineering and Systems Science
24
+ - Mathematics
25
+ - Physics
26
+ - Quantitative Biology
27
+ - Quantitative Finance
28
+ - Statistics
29
+
30
+ Everything I'll tell you will be about these eight fields.
31
+
32
+ How to use me:
33
+ 1. Give me paper's title and (if you have one) it's abstract
34
+ 2. Choose one of two classification modes:
35
+ - Best prediction: Shows the most likely to be true field
36
+ - Top 95%: Shows multiple fields until I'm at least 95% confident that the correct one is among them
37
+ 3. Press the 'Get prediction' button
38
+ 4. Wait for me to tell you which fields of study this paper relates to
39
+ """
40
+ st.markdown(intro_text)
41
+
42
+ # Call the function to display the introduction
43
+ display_intro()
44
+
45
+
46
  @st.cache_resource
47
  def load_model_and_tokenizer():
48
  model_name = 'distilbert/distilbert-base-cased'