Vivien Chappelier commited on
Commit
1dc9286
Β·
1 Parent(s): 174cd15

Default to a model that doesn't need a special authorization

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +1 -1
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: Llama2 Watermarking
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="meta-llama/Llama-2-7b-chat-hf", help='Language model')
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