fixing errors

app.py

@@ -1,6 +1,7 @@
 import logging
 import os
 import string
+import re
 
 import streamlit as st
 from streamlit import session_state
@@ -17,12 +18,18 @@ CONFIGS_PATH = "configs.yaml"
 MAX_TRIES = 6
 CATEGORIES = ["Country", "Animal", "Food", "Movie"]
 
+GEMMA_WORD_PATTERNS = [
+    "(?<=\*)(.*?)(?=\*)",
+    '(?<=")(.*?)(?=")',
+]
+
 configs = {
-    "generation_config":
+    "generation_config": {
         "max_output_tokens": 256,
         "temperature": 1,
         "top_p": 1,
         "top_k": 32,
+    },
     "os_model": "google/gemma-2b-it",
     "device": "cpu",
 }