Update app.py

app.py

@@ -52,7 +52,9 @@ def extract_text(file):
 
 # Urdu content generation model (replace with your preferred model)
 urdu_gen_model_name = "mwz/UrduGPT2"  # Replace placeholder
-urdu_gen_pipe = pipeline("text-generation", model=urdu_gen_model_name)
+
+# Load the model using TensorFlow weights
+urdu_gen_pipe = pipeline("text-generation", model=urdu_gen_model_name, framework="tf")
 
 
 def answer_questions(document, question):
@@ -90,4 +92,4 @@ def main():
                 st.warning("Couldn't find an answer. Try rephrasing your question.")
 
 if __name__ == "__main__":
-    main()
+    main()