JaviSwift commited on
Commit
e464631
1 Parent(s): 11f14a3

New model: Falcon40

Browse files

Falcon180B was too heavy

Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
  import os
5
 
6
- MODEL_NAME = "tiiuae/falcon-180B"
7
 
8
  access_token = os.getenv("HF_ACCESS_TOKEN")
9
  tokenizer = AutoTokenizer.from_pretrained(
@@ -43,10 +43,10 @@ def improve_code(code: str) -> str:
43
  app = gr.Blocks()
44
 
45
  with app:
46
- gr.Markdown("## Servidor MCP para mejora de c贸digo con Falcon 180B")
47
- code_input = gr.Textbox(label="C贸digo original", lines=15)
48
- improve_btn = gr.Button("Mejorar c贸digo")
49
- code_output = gr.Textbox(label="C贸digo mejorado", lines=15)
50
  improve_btn.click(improve_code, inputs=code_input, outputs=code_output)
51
 
52
  app.launch()
 
3
  import torch
4
  import os
5
 
6
+ MODEL_NAME = "tiiuae/falcon-40b"
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 Falcon 40B")
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)
50
  improve_btn.click(improve_code, inputs=code_input, outputs=code_output)
51
 
52
  app.launch()