Update main.py

main.py

@@ -1,12 +1,20 @@
 from flask import Flask, jsonify, render_template, request, make_response
-import requests
-import transformers
+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)
+
+# Your inference code here
+
+
 @app.route("/", methods=["GET", "POST"])
 def index():
     return "Hello S4!"