tomy007 commited on
Commit
59eef8e
·
verified ·
1 Parent(s): 62b3966

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -484,8 +484,11 @@ def filter_input_with_llama_guard(user_input, model="llama-guard-3-8b"):
484
  model=model,
485
  )
486
  # Return the filtered input
 
 
487
  return response.choices[0].message.content.strip()
488
  except Exception as e:
 
489
  print(f"Error with Llama Guard: {e}")
490
  return None
491
 
 
484
  model=model,
485
  )
486
  # Return the filtered input
487
+ st.write(response);
488
+ st.write(response.choices[0].message.content.strip());
489
  return response.choices[0].message.content.strip()
490
  except Exception as e:
491
+ st.write("Error with Llama Guard: {e}");
492
  print(f"Error with Llama Guard: {e}")
493
  return None
494