Update app.py

app.py

@@ -30,6 +30,17 @@ process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=s
 
 time.sleep(5)  # Wait for the server to start
 
+f = h5py.File("best_model_2.h5", mode="r+")
+model_config_string = f.attrs.get("model_config")
+if model_config_string.find('"groups": 1,') != -1:
+    model_config_string = model_config_string.replace('"groups": 1,', '')
+    f.attrs.modify('model_config', model_config_string)
+    f.flush()
+    model_config_string = f.attrs.get("model_config")
+    assert model_config_string.find('"groups": 1,') == -1
+
+f.close() 
+
  # Function to interact with the Ollama model
 
 def query_model(input_text):