Komal-patra commited on
Commit
667ab64
·
verified ·
1 Parent(s): f95e467
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -6,11 +6,9 @@ from peft import PeftModel, PeftConfig
6
 
7
  # Hugging Face login
8
  token = os.environ.get("token")
9
- if token:
10
- login(token)
11
- print("Login is successful")
12
- else:
13
- print("Token not found. Please set your token in the environment variables.")
14
 
15
  # Model and tokenizer setup
16
  MODEL_NAME = "google/flan-t5-base"
 
6
 
7
  # Hugging Face login
8
  token = os.environ.get("token")
9
+ login(token)
10
+ print("login is succesful")
11
+ max_length=512
 
 
12
 
13
  # Model and tokenizer setup
14
  MODEL_NAME = "google/flan-t5-base"