Canstralian commited on
Commit
966401a
·
verified ·
1 Parent(s): a5165ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
  import subprocess
4
 
5
  # Load your Hugging Face model
6
- model_name = "Canstralian/RedTeamAI-new" # Replace with your Hugging Face model path
7
  chatbot = pipeline("text-generation", model=model_name)
8
 
9
  def generate_response(prompt):
 
3
  import subprocess
4
 
5
  # Load your Hugging Face model
6
+ model_name = "Canstralian/RedTeamAI" # Replace with your Hugging Face model path
7
  chatbot = pipeline("text-generation", model=model_name)
8
 
9
  def generate_response(prompt):