Hugging Face's logo

Spaces:
yakine
/
TextGen
Sleeping

App Files Community
yakine commited on
Commit
e22ecd8
·
verified ·
1 Parent(s): a7215a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  app = FastAPI()
7
 
8
  # Load your fine-tuned model and tokenizer
9
- MODEL_NAME = "'aubmindlab/aragpt2-medium'"
10
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
11
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
12
 
 
6
  app = FastAPI()
7
 
8
  # Load your fine-tuned model and tokenizer
9
+ MODEL_NAME = "aubmindlab/aragpt2-medium"
10
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
11
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
12