main.py

from flask import Flask, jsonify, render_template, request, make_response
from transformers import AutoModel, AutoTokenizer


app = Flask(__name__)



# Replace 'your-username' and 'your-repository-name' with your Hugging Face username and repository name
model_name = 'kumar9/word-auto-filled'

# Load the model and tokenizer
model = AutoModel.from_pretrained(model_name)

@app.route("/", methods=["GET", "POST"])
def index():
    return "Hello S4!"