Jyotiyadav commited on
Commit
d5ce0ad
·
verified ·
1 Parent(s): 339e32c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def generate_output(input_text,model):
57
  output = ""
58
  # Initialize the FastLanguageModel
59
  model, tokenizer = FastLanguageModel.from_pretrained(
60
- model_name = "DataIntelligenceTeam/NER-gemma-7b-bnb-4bit",
61
  max_seq_length = 2500,
62
  dtype = None,
63
  load_in_4bit = True,
 
57
  output = ""
58
  # Initialize the FastLanguageModel
59
  model, tokenizer = FastLanguageModel.from_pretrained(
60
+ model_name = "DataIntelligenceTeam/NER-Phi-3-mini-4k-instruct",
61
  max_seq_length = 2500,
62
  dtype = None,
63
  load_in_4bit = True,