Update main.py

main.py

@@ -15,7 +15,7 @@ nltk.download('punkt')
 tokenizer = nltk.data.load('tokenizers/punkt/english.pickle')
 
 HF_TOKEN = getenv("HF_TOKEN")
-MODEL = "mistralai/Mistral-7B-Instruct-v0.2"
+MODEL = "google/gemma-1.1-7b-it"
 FALLBACK_MODELS = [
     "mistralai/Mixtral-8x7B-Instruct-v0.1",
     "mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mistral-7B-Instruct-v0.1"
@@ -63,6 +63,7 @@ async def generate_response(data: InputData) -> Any:
 
         if isinstance(repaired_response, str):
             raise HTTPException(status_code=500, detail="Invalid response from model")
+        
         else:
             cleaned_response = {}
             for key, value in repaired_response.items():