Hugging Face's logo

Spaces:
Yoxas
/
testchatbot
Runtime error

App Files Community
Yoxas commited on
Commit
4941cda
·
verified ·
1 Parent(s): eee45f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
27
  model_response = AutoModelForCausalLM.from_pretrained("openai-community/gpt2").to(device)
28
 
29
  # Load the NLU model for intent detection
30
- nlu_model = AutoModelForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased-finetuned-sst-2-english").to(device)
31
 
32
  # Define the function to find the most relevant document
33
  @spaces.GPU(duration=120)
 
27
  model_response = AutoModelForCausalLM.from_pretrained("openai-community/gpt2").to(device)
28
 
29
  # Load the NLU model for intent detection
30
+ nlu_model = AutoModelForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased-finetuned-sst-2-english")
31
 
32
  # Define the function to find the most relevant document
33
  @spaces.GPU(duration=120)