abidlabs HF Staff commited on
Commit
fa77bfc
·
verified ·
1 Parent(s): 9e3fbd0

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +3 -1
run.py CHANGED
@@ -2,7 +2,9 @@ import gradio as gr
2
 
3
  def calculator(num1, operation, num2):
4
  if operation == "add":
5
- return ValueError("adding is not supported")
 
 
6
  elif operation == "subtract":
7
  return num1 - num2
8
  elif operation == "multiply":
 
2
 
3
  def calculator(num1, operation, num2):
4
  if operation == "add":
5
+ if num2 == 0:
6
+ return ValueError("adding is not supported")
7
+ return num1 + num2
8
  elif operation == "subtract":
9
  return num1 - num2
10
  elif operation == "multiply":