Update README.md

README.md

@@ -52,20 +52,19 @@ The model is open source under apache 2.0. License
 
 
 from pip_library_parser import CodeToDocGenerator
-from atlassian import Jira
 
-import torch
-torch.set_default_device("cuda")
+# Replace 'your_module' and 'YourModule' with the actual module and module name
+module_name = 'your_module'
+module = __import__(module_name)
 
 # Instantiate the CodeToDocGenerator
 generator = CodeToDocGenerator()
 
 # Generate docstrings for the module's functions and methods
-module = Jira
-module_name = "atlassian.Jira"
-
 docs = generator.generate_module_docs(module, module_name)
-print(docs)
+
+# 'docs' now contains a dictionary mapping function/method names to their generated docstrings
+
 ```
 
 ```python