Moran commited on
Commit
99218ae
·
1 Parent(s): cd81eeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -1,7 +1,9 @@
1
  import streamlit as st
2
-
3
- from transformers import AutoTokenizer, AutoModelWithLMHead
4
- from fastai import load_learner
 
 
5
 
6
  #tokenizer = AutoTokenizer.from_pretrained("Moran/Moran_Aviv_Bart")
7
  @st.cache
@@ -21,7 +23,7 @@ text = st.text_area('Enter or paste your text')
21
  st.button('Summarize')
22
 
23
 
24
-
25
  #summary = model(text)
26
  #st.title(summary)
27
 
 
1
  import streamlit as st
2
+ from huggingface_hub import from_pretrained_fastai
3
+ repo_id = "Aviv/Moran_Aviv_Bart"
4
+ inf_learn = from_pretrained_fastai(repo_id)
5
+ #from transformers import AutoTokenizer, AutoModelWithLMHead
6
+ #from fastai import load_learner
7
 
8
  #tokenizer = AutoTokenizer.from_pretrained("Moran/Moran_Aviv_Bart")
9
  @st.cache
 
23
  st.button('Summarize')
24
 
25
 
26
+ st.title(inf_lear_summarize(text))
27
  #summary = model(text)
28
  #st.title(summary)
29