philschmid HF staff commited on
Commit
63c889a
1 Parent(s): dc74d38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -16,7 +16,10 @@ from recommender.main import get_tgi_config
16
 
17
 
18
  def greet(model_id, gpu_memory, num_gpus):
19
- configs = get_tgi_config(model_id, gpu_memory, num_gpus)
 
 
 
20
  if configs is None:
21
  return json.dumps({"error": f"Couldn't generate TGI config for {model_id}"})
22
  return json.dumps(configs)
 
16
 
17
 
18
  def greet(model_id, gpu_memory, num_gpus):
19
+ try:
20
+ configs = get_tgi_config(model_id, gpu_memory, num_gpus)
21
+ except Exception as e:
22
+ return json.dumps({"error": str(e)})
23
  if configs is None:
24
  return json.dumps({"error": f"Couldn't generate TGI config for {model_id}"})
25
  return json.dumps(configs)