gpcerv commited on
Commit
ca8f36a
1 Parent(s): 9320e3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,6 +10,7 @@ pipeline = transformers.pipeline(
10
  model=model,
11
  torch_dtype=torch.float16,
12
  device_map="auto",
 
13
  )
14
 
15
  sequences = pipeline(
@@ -19,7 +20,7 @@ sequences = pipeline(
19
  num_return_sequences=1,
20
  eos_token_id=tokenizer.eos_token_id,
21
  max_length=200,
22
- use_auth_token=os.environ.get('api')
23
  )
24
  for seq in sequences:
25
  print(f"Result: {seq['generated_text']}")
 
10
  model=model,
11
  torch_dtype=torch.float16,
12
  device_map="auto",
13
+ use_auth_token=os.environ.get('api'),
14
  )
15
 
16
  sequences = pipeline(
 
20
  num_return_sequences=1,
21
  eos_token_id=tokenizer.eos_token_id,
22
  max_length=200,
23
+ use_auth_token=os.environ.get('api'),
24
  )
25
  for seq in sequences:
26
  print(f"Result: {seq['generated_text']}")