Update app.py
Browse files
app.py
CHANGED
|
@@ -96,10 +96,11 @@ def respond(
|
|
| 96 |
yield outputs
|
| 97 |
if token == llama.token_eos():
|
| 98 |
break
|
| 99 |
-
|
| 100 |
except Exception as e:
|
| 101 |
# Custom exception handling
|
| 102 |
raise CustomExceptionHandling(e, sys) from e
|
|
|
|
| 103 |
|
| 104 |
|
| 105 |
|
|
|
|
| 96 |
yield outputs
|
| 97 |
if token == llama.token_eos():
|
| 98 |
break
|
| 99 |
+
return outputs
|
| 100 |
except Exception as e:
|
| 101 |
# Custom exception handling
|
| 102 |
raise CustomExceptionHandling(e, sys) from e
|
| 103 |
+
return None
|
| 104 |
|
| 105 |
|
| 106 |
|