Fix API return type annotation to map outputs properly as Textbox

app.py

@@ -1,6 +1,7 @@
 import os
 import time
 import torch
+from typing import Generator
 from fastapi.responses import HTMLResponse
 from gradio import Server
 
@@ -321,7 +322,7 @@ if has_spaces:
         temperature: float = 1.0,
         top_p: float = 0.95,
         max_tokens: int = 4096
-    ):
+    ) -> Generator[str, None, None]:
         if is_fallback:
             yield from simulate_inference(message)
         else:
@@ -334,7 +335,7 @@ else:
         temperature: float = 1.0,
         top_p: float = 0.95,
         max_tokens: int = 4096
-    ):
+    ) -> Generator[str, None, None]:
         if is_fallback:
             yield from simulate_inference(message)
         else: