shorecode commited on
Commit
1081558
·
verified ·
1 Parent(s): a19fec0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -9,10 +9,9 @@ from huggingface_hub import login
9
 
10
  @spaces.GPU
11
  def fine_tune_model(model_name):
12
- try:
13
- #login(api_key.strip())
14
- # Load the model and tokenizer
15
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2)
16
 
17
  return 'WORKS!'#model
18
 
 
9
 
10
  @spaces.GPU
11
  def fine_tune_model(model_name):
12
+ #login(api_key.strip())
13
+ # Load the model and tokenizer
14
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2)
 
15
 
16
  return 'WORKS!'#model
17