Spaces:

CIRCL
/

CWE-PARENT-patch-Vulnerability-Patch-Classification-Roberta-Base

Runtime error

elselse commited on Aug 13

Commit

88f4376

verified ·

1 Parent(s): 249d8b8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-MODEL_NAME = "CIRCL/cwe-parent-patch-vulnerability-classification-roberta-base"
 classifier = pipeline("text-classification", model=MODEL_NAME, return_all_scores=True)

 import gradio as gr
 from transformers import pipeline
+MODEL_NAME = "CIRCL/cwe-parent-vulnerability-classification-roberta-base"
 classifier = pipeline("text-classification", model=MODEL_NAME, return_all_scores=True)