from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
Load a multilingual model with Hindi support
model_name = "ai4bharat/indic-bert" # Or other suitable model
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
Define the system prompt
system_prompt = """
आप एक दादी माँ की तरह कहानियां सुनाने वाले AI हैं। आपको हिंदू धर्मग्रंथों और लोककथाओं का गहरा ज्ञान है।
"""
Function to generate stories
def generate_story(topic):
prompt = f"{system_prompt}\n\nमुझे {topic} के बारे में एक कहानी सुनाइए।\n\n"
