Update app.py

app.py

@@ -16,7 +16,10 @@ from recommender.main import get_tgi_config
 
 
 def greet(model_id, gpu_memory, num_gpus):
-    configs = get_tgi_config(model_id, gpu_memory, num_gpus)
+    try:
+        configs = get_tgi_config(model_id, gpu_memory, num_gpus)
+    except Exception as e:
+        return json.dumps({"error": str(e)})
     if configs is None:
         return json.dumps({"error": f"Couldn't generate TGI config for {model_id}"})
     return json.dumps(configs)