Update main.py

main.py

@@ -1,17 +1,24 @@
 from flask import Flask, jsonify, render_template, request, make_response
-from transformers import AutoModel, AutoTokenizer
+import requests
 
 
 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)
+
+API_URL = "https://api-inference.huggingface.co/models/kumar9/word-auto-filled"
+headers = {"Authorization": "Bearer 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"}
+
+def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.json()
+	
+
 
 @app.route("/", methods=["GET", "POST"])
 def index():
-    return "Hello S4!"
+    output = query({
+	"inputs": "I like you. I love you",})
+    return output