Update app.py
Browse files
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']}")
|