aidan-o-brien commited on
Commit
c24d6e9
1 Parent(s): 4948c4f

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -49
app.py DELETED
@@ -1,49 +0,0 @@
1
- from distutils.command.upload import upload
2
- from io import StringIO
3
- import pandas as pd
4
- import streamlit as st
5
-
6
- from transformers import AutoTokenizer, TFAutoModelForQuestionAnswering
7
-
8
-
9
- @st.cache
10
- def load_data(file):
11
- df = pd.read_csv(file, encoding='utf-8', nrows=50)
12
- return df
13
-
14
-
15
- @st.cache
16
- def load_model_tokenizer():
17
- tokenizer_cp = "albert-base-v2"
18
- model_cp = "aidan-o-brien/recipe-improver"
19
- tokenizer = AutoTokenizer.from_pretrained(tokenizer_cp)
20
- model = TFAutoModelForQuestionAnswering.from_pretrained(model_cp)
21
- return model, tokenizer
22
-
23
-
24
- # Page config
25
- title = "Recipe Improver"
26
- icon = "🍣"
27
- st.set_page_config(page_title=title, page_icon=icon)
28
- st.title(title)
29
-
30
- # Load csv
31
- uploaded_file = st.file_uploader("Choose a csv file", type="csv", key='file_uploader')
32
- if uploaded_file is not None:
33
- df = load_data(uploaded_file)
34
- st.write(df.head())
35
-
36
- # Load tokenizer and model
37
- model, tokenizer = load_model_tokenizer()
38
- st.write("Model and tokenizer successfully loaded.")
39
-
40
- # Pre-process data from csv file
41
-
42
-
43
- # Run inference
44
-
45
-
46
- # Post-process output of model
47
-
48
-
49
- # Present results