Spaces:
Paused
Paused
from transformers import AutoModelForCausalLM, AutoTokenizer | |
# Load the tokenizer and model | |
tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.3") | |
model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.3") | |
# Prepare your input | |
input_text = "What is the most similar item to 'Pepper - Habanero Pepper'?" | |
inputs = tokenizer(input_text, return_tensors="pt") | |
# Generate output | |
outputs = model.generate(**inputs, max_new_tokens=50) | |
print(tokenizer.decode(outputs[0], skip_special_tokens=True)) | |