Downloads · 30 days
7
35% of all-time downloads
generate_charade_prompt
ANT1P4T1C4/generate_charade_prompt
generate_charade_prompt is a machine learning model from ANT1P4T1C4. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
All-time downloads
20
Public
Parameters
77M
1.5 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin308 MB · 50%
At a glance
- Model type
- t5
- Access
- Public
- Created
- Oct 2, 2024
- Updated
- Oct 4, 2024
- SHA
- c4951eab
Datasets
Model card
From the Hugging Face model README
Simple usage of the model:
from transformers import AutoModel, AutoTokenizer, T5ForConditionalGeneration
model_name = 'ANT1P4T1C4/generate_charade_prompt'
model = T5ForConditionalGeneration.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
def generate_word(difficulty):
input = tokenizer(difficulty, return_tensors="pt")
word = model.generate(input["input_ids"], max_length=10, do_sample=True)
return tokenizer.decode(word[0], skip_special_tokens=True)
print("easy: ", generate_word('easy prompt'))
print("medium: ", generate_word(('medium prompt')))
print("hard: ", generate_word("hard prompt"))
- Type
- t5
- Created
- Oct 2, 2024
- Updated
- Oct 4, 2024