Hamzasajjad38 commited on
Commit
1e6784e
·
verified ·
1 Parent(s): 1a4c1bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import subprocess
5
  import sys
6
  import tempfile
7
  import os
8
- from transformers import T5ForConditionalGeneration, T5Tokenizer
9
  from fpdf import FPDF
10
 
11
  # Download spaCy model at runtime
@@ -16,7 +16,8 @@ subprocess.run(
16
 
17
  MODEL_ID = "Hamzasajjad38/t5-small-qg"
18
  tokenizer = T5Tokenizer.from_pretrained(MODEL_ID)
19
- model = T5ForConditionalGeneration.from_pretrained(MODEL_ID)
 
20
  device = "cuda" if torch.cuda.is_available() else "cpu"
21
  model = model.to(device)
22
 
 
5
  import sys
6
  import tempfile
7
  import os
8
+ from transformers import T5ForConditionalGeneration, AutoTokenizer
9
  from fpdf import FPDF
10
 
11
  # Download spaCy model at runtime
 
16
 
17
  MODEL_ID = "Hamzasajjad38/t5-small-qg"
18
  tokenizer = T5Tokenizer.from_pretrained(MODEL_ID)
19
+ #model = T5ForConditionalGeneration.from_pretrained(MODEL_ID)
20
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, use_fast=True)
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
22
  model = model.to(device)
23