ayushraj2349-2 commited on
Commit
e279ae7
·
verified ·
1 Parent(s): c698a3b

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -29
app.py DELETED
@@ -1,29 +0,0 @@
1
- from transformers import AutoModelForCausalLM, AutoTokenizer
2
- import gradio as gr
3
-
4
- # Load Hugging Face's CodeGen model
5
- model_name = "Salesforce/codegen-2B-multi"
6
- tokenizer = AutoTokenizer.from_pretrained(model_name)
7
- model = AutoModelForCausalLM.from_pretrained(model_name)
8
-
9
- # Function to review Python code
10
- def review_code(code_snippet):
11
- inputs = tokenizer(code_snippet, return_tensors="pt")
12
- outputs = model.generate(**inputs, max_length=512)
13
- reviewed_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
14
- return reviewed_code
15
-
16
- # Gradio UI
17
- def check_code(input_code):
18
- return review_code(input_code)
19
-
20
- # Create a Gradio Interface
21
- interface = gr.Interface(
22
- fn=check_code,
23
- inputs=gr.Textbox(label="Enter Python Code"),
24
- outputs=gr.Textbox(label="Reviewed Code"),
25
- title="AI Code Reviewer"
26
- )
27
-
28
- # Launch the app
29
- interface.launch()