Fix setup.py

setup.py

@@ -1,4 +1,4 @@
-from setuptools import setup, find_packages
+from setuptools import setup
 
 with open("README.md", "r", encoding="utf-8") as readme_file:
     readme = readme_file.read()
@@ -15,12 +15,16 @@
 
 setup(
     name='clipseg',
+    packages=['clipseg'],
+    package_dir={'clipseg': 'models'},
+    package_data={'clipseg': [
+        "../weights/*.pth",
+    ]},
     version='0.0.1',
     url='https://github.com/timojl/clipseg',
     python_requires='>=3.9',
     install_requires=requirements,
     description='This repository contains the code used in the paper "Image Segmentation Using Text and Image Prompts".',
-    packages=find_packages(),
     long_description=readme,
     long_description_content_type="text/markdown",
 )