e-hossam96 commited on
Commit
443ce70
·
1 Parent(s): 1805d22

updated format

Browse files
Files changed (1) hide show
  1. main.py +4 -1
main.py CHANGED
@@ -45,7 +45,10 @@ def load_classifier(model_name: str):
45
  ),
46
  )
47
  return pipeline(
48
- "text-classification", model=model, tokenizer=tokenizer, framework="pt"
 
 
 
49
  )
50
 
51
 
 
45
  ),
46
  )
47
  return pipeline(
48
+ "text-classification",
49
+ model=model,
50
+ tokenizer=tokenizer,
51
+ framework="pt",
52
  )
53
 
54