Samet Yilmaz commited on
Commit
b679c08
1 Parent(s): 14e2513

Output text only

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -48,8 +48,7 @@ def infer(image_url, prompt, progress=gr.Progress(track_tqdm=True)):
48
 
49
  outputs = llm.chat(messages, sampling_params=sampling_params)
50
 
51
- print(outputs[0].outputs[0].text)
52
- return outputs
53
 
54
 
55
  example_images = ["https://picsum.photos/id/237/200/300"]
@@ -65,7 +64,7 @@ css = """
65
  with gr.Blocks(css=css) as demo:
66
  with gr.Column(elem_id="col-container"):
67
  gr.Markdown(f"""
68
- # Text-to-Image Gradio Template
69
  """)
70
 
71
  with gr.Row():
 
48
 
49
  outputs = llm.chat(messages, sampling_params=sampling_params)
50
 
51
+ return outputs[0].outputs[0].text
 
52
 
53
 
54
  example_images = ["https://picsum.photos/id/237/200/300"]
 
64
  with gr.Blocks(css=css) as demo:
65
  with gr.Column(elem_id="col-container"):
66
  gr.Markdown(f"""
67
+ # Mistral Pixtral 12B
68
  """)
69
 
70
  with gr.Row():