metya 1 éve
szülő
commit
8d94444749
5 módosított fájl, 12 hozzáadás és 10 törlés
  1. 1 1
      Dockerfile
  2. 3 3
      app.py
  3. 5 3
      docker-compose.yaml
  4. 3 3
      requirements.txt
  5. BIN
      static/audio.wav

+ 1 - 1
Dockerfile

@@ -10,4 +10,4 @@ RUN pip install -r requirements.txt
 
 COPY . .
 
-ENTRYPOINT [ "flask", "run", "-h", "0.0.0.0"]
+ENTRYPOINT [ "flask", "run", "-h", "0.0.0.0", "--debug"]

+ 3 - 3
app.py

@@ -17,7 +17,7 @@ local_file = 'model_v4_ru.pt'
 
 if not os.path.isfile(local_file):
     torch.hub.download_url_to_file('https://models.silero.ai/models/tts/ru/v4_ru.pt',
-                                   local_file)  
+                                   local_file)
 
 model = torch.package.PackageImporter(local_file).load_pickle("tts_models", "model") # type: ignore
 model.to(device)
@@ -33,7 +33,7 @@ class State:
     emotion = []
     age = []
     prompt = ""
-    generation_text = ""
+    generation_text: str | None = ""
     need_generation = True
     new_audio = False
     need_audio = False
@@ -157,7 +157,7 @@ def generate_text(prompt):
                                 ])
     state.need_audio = True
     app.logger.info("openai generation is done")
-    return response['choices'][0]['message']['content'] # type: ignore
+    return response.choices[0].message.content
 
 def generate_audio(sample_text):
     app.logger.info('starting audio generation')

+ 5 - 3
docker-compose.yaml

@@ -5,14 +5,16 @@ services:
     build: .
     container_name: symetria_emotions
     restart: always
+    volumes:
+      - ./:/app
     expose:
       - 5000
     environment:
-      VIRTUAL_HOST: symetria.chad-partners.com
-      LETSENCRYPT_HOST: symetria.chad-partners.com
+      VIRTUAL_HOST: onism.chad-partners.com
+      LETSENCRYPT_HOST: onism.chad-partners.com
     networks:
       - nginx-proxy
 
 networks:
   nginx-proxy:
-    external: true
+    external: true

+ 3 - 3
requirements.txt

@@ -1,7 +1,7 @@
 --find-links https://download.pytorch.org/whl/torch_stable.html
-Flask==2.3.2
-openai==0.27.8
-torch==2.0.1+cpu
+Flask
+openai
+torch==2.2.1+cpu
 torchvision 
 torchaudio 
 # gunicorn

BIN
static/audio.wav