minjibi commited on
Commit
9cf3d4c
·
1 Parent(s): 935d100

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -10
app.py CHANGED
@@ -7,20 +7,12 @@ from transformers import (
7
  MT5TokenizerFast,
8
  )
9
 
10
- # model = MT5ForConditionalGeneration.from_pretrained(
11
- # "minjibi/qa",
12
- # return_dict=True,
13
- # )
14
- # tokenizer = MT5TokenizerFast.from_pretrained(
15
- # "minjibi/qa"
16
- # )
17
-
18
  model = MT5ForConditionalGeneration.from_pretrained(
19
- "minjibi/autotrain-qa-base-testdf-51358121752",
20
  return_dict=True,
21
  )
22
  tokenizer = MT5TokenizerFast.from_pretrained(
23
- "minjibi/autotrain-qa-base-testdf-51358121752"
24
  )
25
 
26
  def predict(text):
 
7
  MT5TokenizerFast,
8
  )
9
 
 
 
 
 
 
 
 
 
10
  model = MT5ForConditionalGeneration.from_pretrained(
11
+ "minjibi/qa",
12
  return_dict=True,
13
  )
14
  tokenizer = MT5TokenizerFast.from_pretrained(
15
+ "minjibi/qa"
16
  )
17
 
18
  def predict(text):