Hugging Face's logo

Spaces:
DR-Rakshitha
/
wizardlm_api
Runtime error

App Files Community
1
DR-Rakshitha commited on
Commit
d048f50
·
1 Parent(s): 0d5d23d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
11
  model_directory = "./"
12
 
13
  # Load the model and tokenizer
14
- model = AutoModelForCausalLM.from_pretrained(model_directory)
15
  tokenizer = AutoTokenizer.from_pretrained(model_directory, trust_remote_code=True)
16
 
17
  # Now you can generate text as before
 
11
  model_directory = "./"
12
 
13
  # Load the model and tokenizer
14
+ model = AutoModelForCausalLM.from_pretrained(model_directory, from_tf=True)
15
  tokenizer = AutoTokenizer.from_pretrained(model_directory, trust_remote_code=True)
16
 
17
  # Now you can generate text as before