Upload app.py with huggingface_hub
Browse files
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 |
|