rica commited on
Commit
67a650b
1 Parent(s): da54da6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,9 +1,8 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- pipe = pipeline(task="text-classification",
5
- # model that can do 22k-category classification
6
- model="mrm8488/codebert-base-finetuned-detect-insecure-code")
7
  gr.Interface.from_pipeline(pipe,
8
  title="Insecure Code Detector",
9
  description="Label 1 for insecure code",
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipe = pipeline(task="text-classification",
5
+ model="mrm8488/codebert-base-finetuned-detect-insecure-code")
 
6
  gr.Interface.from_pipeline(pipe,
7
  title="Insecure Code Detector",
8
  description="Label 1 for insecure code",