Update app.py

app.py

@@ -180,11 +180,22 @@ def main():
         if st.checkbox("Spell Corrections for English"):
             st.success(TextBlob(text).correct())
         if st.checkbox("Text Generation"):
-            tokenizer, model = load_models()
-            input_ids = tokenizer(text, return_tensors='pt').input_ids
-            st.text("Using Hugging Face Transformer, Contrastive Search ..")
-            output = model.generate(input_ids, max_length=128)
-            st.success(tokenizer.decode(output[0], skip_special_tokens=True))
+            API_URL = "https://api-inference.huggingface.co/models/gpt2-large"
+            headers = {"Authorization": "Bearer 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"}
+            
+            def query(payload):
+            	response = requests.post(API_URL, headers=headers, json=payload)
+            	return response.json()
+            	
+            output = query({
+            	"inputs": text,
+            })
+            st.success(output)
+            # tokenizer, model = load_models()
+            # input_ids = tokenizer(text, return_tensors='pt').input_ids
+            # st.text("Using Hugging Face Transformer, Contrastive Search ..")
+            # output = model.generate(input_ids, max_length=128)
+            #st.success(tokenizer.decode(output[0], skip_special_tokens=True))
         # if st.checkbox("Mark here, Text Summarization for English or Bangla!"):
         #     st.subheader("Summarize Your Text for English and Bangla Texts!")
         #     message = st.text_area("Enter the Text","Type please ..")