Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
 # Load the model and tokenizer from Hugging Face repository
-model_repo_id = "Ajay12345678980/QA_bot"  # Your model repository
+model_repo_id = "Ajay12345678980/QA_bot"  # Replace with your model repository ID
 
 # Initialize the model and tokenizer
 model = GPT2LMHeadModel.from_pretrained(model_repo_id)
@@ -23,15 +23,7 @@ def predict(text):
         # Decode the generated output
         prediction = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-        # Extract the text between "Answer" and "ANS" using regex
-        match = re.search(r'Answer\s*(.*?)\s*<ANS', prediction, re.DOTALL)
-        
-        if match:
-            # Return the extracted content
-            return match.group(1).strip()
-        else:
-            # If no match is found, return a message indicating that
-            return "No answer found between 'Answer' and '<ANS>' markers."
+        return prediction.strip()  # Return the clean output
     except Exception as e:
         # Handle and print any exceptions for debugging
         return f"An error occurred: {str(e)}"