Update app.py

app.py

@@ -1,11 +1,14 @@
+import re
+import torch
 import gradio as gr
-from transformers import pipeline
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-# Load the model using the Hugging Face pipeline
-model = pipeline("text-generation", model="Ajay12345678980/QA_bot")
+# Load the model and tokenizer from Hugging Face repository
+model_repo_id = "Ajay12345678980/QA_bot"  # Your model repository
 
-import re
-import torch
+# Initialize the model and tokenizer
+model = GPT2LMHeadModel.from_pretrained(model_repo_id)
+tokenizer = GPT2Tokenizer.from_pretrained(model_repo_id)
 
 # Define the prediction function
 def predict(text):
@@ -45,4 +48,3 @@ interface = gr.Interface(
 # Launch the Gradio app
 if __name__ == "__main__":
     interface.launch()
-