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 |
|