edwardb1203 commited on
Commit
b6b30a5
1 Parent(s): d44dac4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ import gradio as gr
5
  import os
6
  import docx2txt
7
 
8
- tokenizer = LEDTokenizer.from_pretrained("patrickvonplaten/led-large-16384-pubmed")
9
- model = LEDForConditionalGeneration.from_pretrained("patrickvonplaten/led-large-16384-pubmed").to("cuda").half()
10
 
11
  def summarize(text_file):
12
  file_extension = os.path.splitext(text_file.name)[1]
 
5
  import os
6
  import docx2txt
7
 
8
+ tokenizer = LEDTokenizer.from_pretrained("nalmy/abstract-comp590")
9
+ model = LEDForConditionalGeneration.from_pretrained("nalmy/abstract-comp590").to("cuda").half()
10
 
11
  def summarize(text_file):
12
  file_extension = os.path.splitext(text_file.name)[1]