DeepMount00
/

OCR_corrector

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

DeepMount00 commited on Apr 10

Commit

37abb7c

•

1 Parent(s): 6117df9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model.to(device)
 my_text = ""
 inputs = tokenizer(my_text, return_tensors="pt").to(device)
-outputs = finetuned_model.generate(input_ids=inputs['input_ids'],
                attention_mask=inputs['attention_mask'],
                num_beams=2, max_length=1050, top_k=10)
 clean_text = tokenizer.decode(outputs[0], skip_special_tokens=True)

 model.to(device)
 my_text = ""
 inputs = tokenizer(my_text, return_tensors="pt").to(device)
+outputs = model.generate(input_ids=inputs['input_ids'],
                attention_mask=inputs['attention_mask'],
                num_beams=2, max_length=1050, top_k=10)
 clean_text = tokenizer.decode(outputs[0], skip_special_tokens=True)