Hugging Face's logo

Spaces:
Flexy0
/
VeoAI
Runtime error

App Files Community
Flexy0 commited on
Commit
b962a3e
•
1 Parent(s): b6b3125

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, DataCollatorForSeq2Seq, Seq2SeqTrainer, Seq2SeqTrainingArguments
2
  # Load tokenizer and model
3
- tokenizer = AutoTokenizer.from_pretrained("gpt2-large")
4
  model = AutoModelForSeq2SeqLM.from_pretrained("Kaludi/chatgpt-gpt4-prompts-bart-large-cnn-samsum", from_tf=True)
5
 
6
  # Assuming you have your own dataset for fine-tuning
 
1
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, DataCollatorForSeq2Seq, Seq2SeqTrainer, Seq2SeqTrainingArguments
2
  # Load tokenizer and model
3
+ tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2-large")
4
  model = AutoModelForSeq2SeqLM.from_pretrained("Kaludi/chatgpt-gpt4-prompts-bart-large-cnn-samsum", from_tf=True)
5
 
6
  # Assuming you have your own dataset for fine-tuning