Spaces:
Runtime error
Runtime error
Vivien Chappelier
commited on
Commit
Β·
1dc9286
1
Parent(s):
174cd15
Default to a model that doesn't need a special authorization
Browse files
README.md
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
---
|
2 |
-
title:
|
3 |
emoji: π
|
4 |
colorFrom: green
|
5 |
colorTo: indigo
|
|
|
1 |
---
|
2 |
+
title: Text Watermarking
|
3 |
emoji: π
|
4 |
colorFrom: green
|
5 |
colorTo: indigo
|
app.py
CHANGED
@@ -13,7 +13,7 @@ hf_token = os.getenv('HF_TOKEN')
|
|
13 |
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
14 |
|
15 |
parser = argparse.ArgumentParser(description='Generative Text Watermarking demo')
|
16 |
-
parser.add_argument('--model', '-m', type=str, default="
|
17 |
parser.add_argument('--key', '-k', type=int, default=42,
|
18 |
help='The seed of the pseudo random number generator')
|
19 |
|
|
|
13 |
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
14 |
|
15 |
parser = argparse.ArgumentParser(description='Generative Text Watermarking demo')
|
16 |
+
parser.add_argument('--model', '-m', type=str, default="microsoft/Phi-3-mini-4k-instruct", help='Language model')
|
17 |
parser.add_argument('--key', '-k', type=int, default=42,
|
18 |
help='The seed of the pseudo random number generator')
|
19 |
|