compare-security-models

Running on Zero

nyasukun commited on 22 days ago

Commit

3119f87

1 Parent(s): fc804b6

.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model_options = {
     "Foundation-Sec-8B": pipeline("text-generation", model="fdtn-ai/Foundation-Sec-8B"),
 }
-#@spaces.GPU
 def generate_text_local(model_pipeline, prompt):
     """Local text generation"""
     try:

     "Foundation-Sec-8B": pipeline("text-generation", model="fdtn-ai/Foundation-Sec-8B"),
 }
+@spaces.GPU
 def generate_text_local(model_pipeline, prompt):
     """Local text generation"""
     try: