Go-Gourmet Fine-Tuned Mistral 7B Model
Overview
The Go-Gourmet model is a fine-tuned version of the mistralai/Mistral-7B-Instruct-v0.2 base model with 32k context window, specifically trained to generate structured restaurant cards based on an input of a restaurant name and location. The model has been fine-tuned using a custom dataset of restaurant information to capture relevant details such as cuisine, opening times, location, rating, average price, best dishes, pre-booking requirements, dress code, and website.
Model Details
- Base Model:
mistralai/Mistral-7B-Instruct-v0.2. - Fine-Tuning Dataset: Custom dataset of restaurant information
Sadiah/Go-Gourmet. - Fine-Tuning Approach:
QLoRAandSFT Trainer. - Model Size: The model retains the same size and architecture as the original Mistral base model.
Intended Use
The Go-Gourmet Fine-Tuned Mistral Model is designed to generate structured restaurant cards based on an input of a restaurant name and location. It can be used for various purposes, such as:
- Generating informative restaurant cards for food and travel applications
- Providing quick and structured information about restaurants to users
- Enhancing natural language processing applications related to the food and hospitality industry
Limitations and Considerations
- The model's outputs are generated based on patterns and characteristics learned from the fine-tuning dataset. While it aims to provide accurate and relevant information, the generated restaurant cards may not always be perfect or up-to-date.
- The model relies on the quality and comprehensiveness of the fine-tuning dataset. If certain details or categories are missing from the dataset, the model may not be able to generate them accurately.
- The generated restaurant cards should be used as a starting point and should be verified with official sources or the restaurants themselves for the most accurate and current information.
Inference Code
To test and interact with the Go-Gourmet Fine-Tuned Mistral Model, you can use the following inference code:
# Load the fine-tuned model from hugging face
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
tokenizer_new = AutoTokenizer.from_pretrained("Sadiah/Go-Gourmet")
model_new = AutoModelForCausalLM.from_pretrained(
"Sadiah/Go-Gourmet",
trust_remote_code=True,
torch_dtype=torch.float16,
device_map= {"": 0},
)
input_text = '''[INST]Olives, Delhi [/INST]''' #Define instruction
input_ids = tokenizer_new(input_text, return_tensors="pt") #Tokenize instruction
input_ids = input_ids.to("cuda") #Move instruction to GPU
outputs = model_new.generate(**input_ids, max_length=300, num_return_sequences=1, temperature=0) #Generate response
generated_text = tokenizer_new.decode(outputs[0]) #Decode generated response
# Find the index of the closing instruction tag and remove the instruction
instruction_end_index = generated_text.find("[/INST]")
if instruction_end_index != -1:
generated_text = generated_text[instruction_end_index + len("[/INST]"):].strip()
# Find the index of "Website:" and the end of the website address
website_start_index = generated_text.find("Website:")
if website_start_index != -1:
website_end_index = generated_text.find("\n", website_start_index)
if website_end_index == -1:
website_end_index = len(generated_text)
truncated_text = generated_text[:website_end_index]
print(truncated_text.strip())
else:
print(generated_text)
Name: Olives, Delhi
Cuisine: Mediterranean
Opening Times: Mon-Sun: 12:30pm-3:30pm, 7pm-11:30pm
Location: 1, Kalka Das Marg, New Delhi, Delhi 110001, India (28.7031, 77.1123)
Rating: 4.3 (Source: Zomato)
Average Price Per Person: Moderate
Three Best Dishes:
1. Grilled Halloumi Cheese: A popular Mediterranean cheese, grilled to perfection.
2. Falafel Platter: A selection of crispy, flavorful falafel balls served with hummus and pita bread.
3. Lamb Shank Tagine: Slow-cooked lamb shank in a rich, aromatic sauce.
Pre-Booking Needed: Recommended, especially for dinner
Dress Code: Casual
Website: http://www.olivesdelhi.com/
This code snippet allows you to provide an input of a restaurant name and location and generate a structured restaurant card using the Go-Gourmet Fine-Tuned Mistral Model.
Contact and Feedback
If you have any questions, feedback, or concerns regarding the Go-Gourmet Fine-Tuned Mistral Model, please contact me at https://www.sadiahzahoor.com/contact .