codelion commited on
Commit
54683ea
ยท
1 Parent(s): cfb47a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,9 +4,9 @@ from transformers import pipeline
4
  import os
5
  import torch
6
 
7
- description = """# <p style="text-align: center; color: white;"> ๐ŸŽ… <span style='color: #ff75b3;'>SantaFixer:</span> Code Generation </p>
8
- <span style='color: white;'>This is a demo to generate code with <a href="https://huggingface.co/bigcode/santacoder" style="color: #ff75b3;">SantaCoder</a>,
9
- a 1.1B parameter model for code generation in Python, Java & JavaScript. The model can also do infilling, just specify where you would like the model to complete code
10
  with the <span style='color: #ff75b3;'>&lt;FILL-HERE&gt;</span> token.</span>"""
11
 
12
  # token = os.environ["HUB_TOKEN"]
 
4
  import os
5
  import torch
6
 
7
+ description = """# <p style="text-align: center; color: white;"> ๐ŸŽ…๐Ÿ”จ <span style='color: #ff75b3;'>SantaFixer:</span> Bug Fixing</p>
8
+ <span style='color: white;'>This is a demo to generate code with <a href="https://huggingface.co/lambdasec/santafixer" style="color: #ff75b3;">SantaFixer</a>,
9
+ a fine-tuned SantaCoder model that is focussed on bug fixing in Python, Java & JavaScript using infilling, just specify where you would like the model to fix code
10
  with the <span style='color: #ff75b3;'>&lt;FILL-HERE&gt;</span> token.</span>"""
11
 
12
  # token = os.environ["HUB_TOKEN"]