Spaces:

pushkarraj
/

opt13b_paraphraser

Runtime error

pushkarraj commited on Dec 27, 2022

Commit

345ff01

1 Parent(s): b6916df

quotes added

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,8 @@ import torch
 from transformers import pipeline, GPT2Tokenizer, OPTForCausalLM
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model=OPTForCausalLM.from_pretrained('pushkarraj/opt_paraphraser')
-tokenizer=GPT2Tokenizer.from_pretrained('pushkarraj/opt_paraphraser',truncation=True)
 generator=pipeline("text-generation",model=model,tokenizer=tokenizer,device=device)

 from transformers import pipeline, GPT2Tokenizer, OPTForCausalLM
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model=OPTForCausalLM.from_pretrained("pushkarraj/opt_paraphraser")
+tokenizer=GPT2Tokenizer.from_pretrained("pushkarraj/opt_paraphraser",truncation=True)
 generator=pipeline("text-generation",model=model,tokenizer=tokenizer,device=device)