Hugging Face's logo

DisgustingOzil
/
Academic-ShortQA-Generator

Transformers
Safetensors
unsloth
Model card Files
xet
 Community
DisgustingOzil commited on
Commit
bcbfbad
·
verified ·
1 Parent(s): 844d72c

updated handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
  class EndpointHandler:
5
  def __init__(self, path=""):
6
- model_id = "DisgustingOzil/Academic-MCQ-Generator"
7
  load_in_4bit = True
8
  self.tokenizer = AutoTokenizer.from_pretrained(model_id)
9
  self.model = AutoModelForCausalLM.from_pretrained(model_id, load_in_4bit=load_in_4bit)
 
3
 
4
  class EndpointHandler:
5
  def __init__(self, path=""):
6
+ model_id = "DisgustingOzil/Academic-ShortQA-Generator"
7
  load_in_4bit = True
8
  self.tokenizer = AutoTokenizer.from_pretrained(model_id)
9
  self.model = AutoModelForCausalLM.from_pretrained(model_id, load_in_4bit=load_in_4bit)