Update app.py

app.py

@@ -17,9 +17,16 @@ async def generate_text():
         
         hf_hub_download(repo_id="TheBloke/Mistral-7B-v0.1-GGUF", filename="mistral-7b-v0.1.Q4_K_M.gguf", local_dir=model_dir)
 
+        # Check if config.json file exists in the model directory
+        config_file = os.path.join(model_dir, "config.json")
+        if not os.path.exists(config_file):
+            raise ValueError("config.json file is missing in the model directory")
+
+        # Load tokenizer and model
         tokenizer = AutoTokenizer.from_pretrained(model_dir)
         model = AutoModelForCausalLM.from_pretrained(model_dir)
 
+        # Generate text
         prompt = "Once upon a time, there was a"
         inputs = tokenizer(prompt, return_tensors="pt")
         output = model.generate(input_ids=inputs["input_ids"], max_length=50, num_return_sequences=3, temperature=0.7)