workflow fixed

.github/workflows/hugging-face.yaml

@@ -1,19 +1,20 @@
-name: Unit Tests
-
+name: Uploading on Huggingface
 on:
   push:
-    branches:
-      - "main"
-# TODO setup for huggingface space
+    branches: [main]
+  workflow_dispatch:
+
 jobs:
-  test:
+  sync-to-hub:
     runs-on: ubuntu-latest
-
     steps:
-      - name: Checkout code
-        uses: actions/checkout@v2
-
-      - name: Setup Python
-        uses: actions/setup-python@v2
+      - uses: actions/checkout@v3
         with:
-          python-version: 3.x
+          fetch-depth: 0
+          lfs: true
+      - name: Push to hub
+        env:
+          HF_TOKEN: ${{ secrets.HF_TOKEN }}
+          HF_USERNAME: zaibutcooler
+          SPACE_NAME: yume
+        run: git push https://${{ secrets.HF_USERNAME }}:$HF_TOKEN@huggingface.co/spaces/${{ secrets.HF_USERNAME }}/${{ secrets.SPACE_NAME }} main

.github/workflows/unit-test.yaml

@@ -10,18 +10,16 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-      - name: Checkout code
+      - name: Checkout repository
         uses: actions/checkout@v2
 
-      - name: Setup Python
+      - name: Set up Python
         uses: actions/setup-python@v2
         with:
           python-version: 3.x
 
       - name: Install dependencies
-        run: |
-          pip install -r ./requirements.txt
+        run: pip install -r requirements.txt
 
       - name: Run tests
-        run: |
-          python -m unittest ./tests
+        run: python -m unittest discover tests

README.md

@@ -1,3 +1,13 @@
+---
+title: {{title}}
+emoji: {{emoji}}
+colorFrom: {{colorFrom}}
+colorTo: {{colorTo}}
+sdk: {{sdk}}
+sdk_version: {{sdkVersion}}
+app_file: app.py
+pinned: false
+---
 # Yume: Image Generation with Diffusion model using PyTorch
 
 ## Overview
@@ -68,5 +78,5 @@ Mention any contributors or libraries that you used or were inspired by.
 ## Contact
 
 - Zai
-- zaiyellyintaung@gmail.com
+- <zaiyellyintaung@gmail.com>
 - Project Link: [https://github.com/zaibutcooler/yume](https://github.com/zaibutcooler/yume)

tests/test_datasets.py

@@ -2,10 +2,10 @@ import unittest
 
 
 class TestDatasets(unittest.TestCase):
-    def test_download():
+    def test_download(self):
         pass
 
-    def test_encode():
+    def test_encode(self):
         pass
 
 

tests/test_demo.py

@@ -1,19 +0,0 @@
-# test_demo.py
-import unittest
-
-
-class TestDemoFunctions(unittest.TestCase):
-
-    def test_add_numbers(self):
-        self.assertEqual(add_numbers(2, 3), 5)
-        self.assertEqual(add_numbers(-1, 1), 0)
-        # Add more test cases as needed
-
-    def test_multiply_numbers(self):
-        self.assertEqual(multiply_numbers(2, 3), 6)
-        self.assertEqual(multiply_numbers(-1, 1), -1)
-        # Add more test cases as needed
-
-
-if __name__ == "__main__":
-    unittest.main()

tests/test_pretrained.py

@@ -2,10 +2,10 @@ import unittest
 
 
 class TestPretrained(unittest.TestCase):
-    def test_download():
+    def test_download(self):
         pass
 
-    def test_generation():
+    def test_generation(self):
         pass
 
 

tests/test_tokenizer.py

@@ -2,13 +2,13 @@ import unittest
 
 
 class TestTokenizer(unittest.TestCase):
-    def test_encode():
+    def test_encode(self):
         pass
 
-    def test_decode():
+    def test_decode(self):
         pass
 
-    def test_equal_result():
+    def test_equal_result(self):
         pass
 
 

tests/test_transformer.py

@@ -2,10 +2,10 @@ import unittest
 
 
 class TestTransformer(unittest.TestCase):
-    def test_loading_model():
+    def test_loading_model(self):
         pass
 
-    def test_parameters():
+    def test_parameters(self):
         pass