Update app.py

app.py

@@ -53,10 +53,11 @@ def generate_text(message, history):
     for out in output:
         stream = copy.deepcopy(out)
         temp += stream["choices"][0]["text"]
-        yield temp
 
     history.append(("USER:", message))
     history.append(("ASSISTANT:", temp))
+    
+    return temp
 
 # Define the predict function for the FastAI model
 def predict_with_llama_and_generate_text(img):
@@ -66,8 +67,9 @@ def predict_with_llama_and_generate_text(img):
     
     response = f"The system has detected {detected_object}. Do you want to know about {detected_object}?"
     
-    for llama_response in generate_text(response, history):
-        yield llama_response
+    llama_response = generate_text(response, history)
+    
+    return llama_response
 
 # Define the Gradio interface
 gr.Interface(