JaviSwift commited on
Commit
9fd3590
·
1 Parent(s): 35c15e0

New model: Falcon3-7B-Base

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
  import os
5
 
6
- MODEL_NAME = "tiiuae/falcon-7b"
7
 
8
  access_token = os.getenv("HF_ACCESS_TOKEN")
9
  tokenizer = AutoTokenizer.from_pretrained(
@@ -43,7 +43,7 @@ def improve_code(code: str) -> str:
43
  app = gr.Blocks()
44
 
45
  with app:
46
- gr.Markdown("## MCP Server Code Improver with Falcon 7B")
47
  code_input = gr.Textbox(label="Original code", lines=15)
48
  improve_btn = gr.Button("Improve Code")
49
  code_output = gr.Textbox(label="Improved code", lines=15)
 
3
  import torch
4
  import os
5
 
6
+ MODEL_NAME = "tiiuae/Falcon3-7B-Base"
7
 
8
  access_token = os.getenv("HF_ACCESS_TOKEN")
9
  tokenizer = AutoTokenizer.from_pretrained(
 
43
  app = gr.Blocks()
44
 
45
  with app:
46
+ gr.Markdown("## MCP Server Code Improver with Falcon3-7B-Base")
47
  code_input = gr.Textbox(label="Original code", lines=15)
48
  improve_btn = gr.Button("Improve Code")
49
  code_output = gr.Textbox(label="Improved code", lines=15)