Upgrade to Mistral-7B-Instruct-v0.2

.idea/misc.xml

@@ -1,4 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.10 (slide-deck-ai)" />
+  </component>
   <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (slide-deck-ai)" project-jdk-type="Python SDK" />
 </project>

app.py

@@ -1,15 +1,11 @@
-import base64
-import os
 import pathlib
-import json5
 import logging
-import shutil
-import time
-import streamlit as st
-import streamlit.runtime.scriptrunner as st_sr
+import tempfile
 from typing import List, Tuple
+
+import json5
 import metaphor_python as metaphor
-import tempfile
+import streamlit as st
 
 import llm_helper
 import pptx_helper
@@ -115,7 +111,7 @@ def build_ui():
 
     st.title(APP_TEXT['app_name'])
     st.subheader(APP_TEXT['caption'])
-    st.markdown('Powered by [Mistral-7B-Instruct-v0.1](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.1).')
+    st.markdown('Powered by [Mistral-7B-Instruct-v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.1).')
     st.markdown('*If the JSON is generated or parsed incorrectly, try again later by making minor changes '
                 'to the input text.*')
 

global_config.py

@@ -8,22 +8,22 @@ load_dotenv()
 
 @dataclass(frozen=True)
 class GlobalConfig:
-    CLARIFAI_PAT = os.environ.get('CLARIFAI_PAT', '')
-    CLARIFAI_USER_ID = 'meta'
-    CLARIFAI_APP_ID = 'Llama-2'
-    CLARIFAI_MODEL_ID = 'llama2-13b-chat'
-
-    CLARIFAI_USER_ID_GPT = 'openai'
-    CLARIFAI_APP_ID_GPT = 'chat-completion'
-    CLARIFAI_MODEL_ID_GPT = 'GPT-4'  # 'GPT-3_5-turbo'
-
-    CLARIFAI_USER_ID_SD = 'stability-ai'
-    CLARIFAI_APP_ID_SD = 'stable-diffusion-2'
-    CLARIFAI_MODEL_ID_SD = 'stable-diffusion-xl'
-    CLARIFAI_MODEL_VERSION_ID_SD = '0c919cc1edfc455dbc96207753f178d7'
-
-    HF_LLM_MODEL_NAME = 'mistralai/Mistral-7B-Instruct-v0.1'  # 'HuggingFaceH4/zephyr-7b-beta'
-    LLM_MODEL_TEMPERATURE: float = 0.2  # Mistral and Zephyr require a bit high temperature
+    # CLARIFAI_PAT = os.environ.get('CLARIFAI_PAT', '')
+    # CLARIFAI_USER_ID = 'meta'
+    # CLARIFAI_APP_ID = 'Llama-2'
+    # CLARIFAI_MODEL_ID = 'llama2-13b-chat'
+    #
+    # CLARIFAI_USER_ID_GPT = 'openai'
+    # CLARIFAI_APP_ID_GPT = 'chat-completion'
+    # CLARIFAI_MODEL_ID_GPT = 'GPT-4'  # 'GPT-3_5-turbo'
+    #
+    # CLARIFAI_USER_ID_SD = 'stability-ai'
+    # CLARIFAI_APP_ID_SD = 'stable-diffusion-2'
+    # CLARIFAI_MODEL_ID_SD = 'stable-diffusion-xl'
+    # CLARIFAI_MODEL_VERSION_ID_SD = '0c919cc1edfc455dbc96207753f178d7'
+
+    HF_LLM_MODEL_NAME = 'mistralai/Mistral-7B-Instruct-v0.2'
+    LLM_MODEL_TEMPERATURE: float = 0.2
     LLM_MODEL_MIN_OUTPUT_LENGTH: int = 50
     LLM_MODEL_MAX_OUTPUT_LENGTH: int = 2000
     LLM_MODEL_MAX_INPUT_LENGTH: int = 300