Update app.py

app.py

@@ -131,22 +131,21 @@ with gr.Blocks() as demo:
             def end(self):
                 queue.put(None)
         def background_fn():
-            result = pipe(
-                prefix,
-                streamer=MyStreamer(), 
-                max_length=3000,
-                temperature=0.95,
-            )
+            try:
+                result = pipe(
+                    prefix,
+                    streamer=MyStreamer(), 
+                    max_length=3000,
+                    temperature=0.95,
+                )
+            except queue.Full:
+                print("Queue is full. Exiting.")
             print('Generated text:')
             print(result[0]['generated_text'])
             print()
         Thread(target=background_fn).start()
         while True:
-            try:
-                text = queue.get()
-            except queue.Full:
-                print("Queue is full. Exiting.")
-                break
+            text = queue.get()
             if text is None:
                 break
             history[-1]["content"] += text