Spaces:

RoAr777
/

code_detector

Runtime error

RoAr777 commited on Nov 14, 2023

Commit

7e2ae5e

1 Parent(s): 18e3a2b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
 tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
-def chatbot_model(prompt):
     # Encode the prompt and generate response
     input_ids = tokenizer.encode(prompt + tokenizer.eos_token, return_tensors='pt')

 model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
 tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
+def chatbot_model(prompt,history):
     # Encode the prompt and generate response
     input_ids = tokenizer.encode(prompt + tokenizer.eos_token, return_tensors='pt')