Update app.py

app.py

@@ -13,26 +13,14 @@ def format_prompt(message, history):
     return prompt
 
 def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=9048, top_p=0.95, repetition_penalty=1.0):
-    temperature = max(float(temperature), 1e-2)
-    top_p = float(top_p)
-
-    generate_kwargs = dict(
-        temperature=temperature,
-        max_new_tokens=max_new_tokens,
-        top_p=top_p,
-        repetition_penalty=repetition_penalty,
-        do_sample=True,
-        seed=42,
-    )
-
-    formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
-    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
-    output = ""
+    # ... existing code ...
 
     for response in stream:
         output += response.token.text
-        yield output
-    return output
+        if "http" in output:  # assuming the AI writes a direct image link in its response
+            yield {"image": output}  # Gradio will display the image
+        else:
+            yield output
 
 additional_inputs = [
     gr.Textbox(label="System Prompt", max_lines=1, interactive=True),
@@ -46,7 +34,7 @@ gr.ChatInterface(
     fn=generate,
     chatbot=gr.Chatbot(show_label=False, show_share_button=True, show_copy_button=True, likeable=True, layout="panel"),
     additional_inputs=additional_inputs,
-    title="FreeAI",
+    title="ConvoLite",
     concurrency_limit=20,
     
 ).launch(show_api=False,)