Spaces:

Sushrut98
/

emotion

Sleeping

Sushrut98 commited on Feb 21, 2024

Commit

cfeafc3

verified ·

1 Parent(s): 31228a5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,7 +39,7 @@ TinyCNN_model.load_state_dict(
 ### 3. Predict function ###
 # Create predict function
-def predict(img):
     """Transforms and performs a prediction on img and returns prediction and time taken.
     """
     # Start the timer

 ### 3. Predict function ###
 # Create predict function
+def predict(img) -> Tuple[Dict, float]:
     """Transforms and performs a prediction on img and returns prediction and time taken.
     """
     # Start the timer