Update app.py (#7)

- Update app.py (381ab5c96ab6b4588fba45cab0769fe9d544c736)

app.py

@@ -11,6 +11,12 @@ import torch
 import spaces
 import requests
 
+CSS ="""
+.contain { display: flex; flex-direction: column; }
+#component-0 { height: 100%; }
+#chatbot { flex-grow: 1; }
+"""
+
 model_id = "xtuner/llava-llama-3-8b-v1_1-transformers"
 
 processor = AutoProcessor.from_pretrained(model_id)
@@ -43,7 +49,7 @@ def bot_streaming(message, history):
   inputs = processor(prompt, image, return_tensors='pt').to(0, torch.float16)
 
   streamer = TextIteratorStreamer(processor, **{"skip_special_tokens": True})
-  generation_kwargs = dict(inputs, streamer=streamer, max_new_tokens=200)
+  generation_kwargs = dict(inputs, streamer=streamer, max_new_tokens=1024)
   generated_text = ""
 
   thread = Thread(target=model.generate, kwargs=generation_kwargs)
@@ -62,7 +68,7 @@ def bot_streaming(message, history):
     yield generated_text_without_prompt
 
 
-demo = gr.ChatInterface(fn=bot_streaming, title="LLaVA Llama-3-8B", examples=[{"text": "What is on the flower?", "files":["./bee.jpg"]},
+demo = gr.ChatInterface(fn=bot_streaming, css=CSS, title="LLaVA Llama-3-8B", examples=[{"text": "What is on the flower?", "files":["./bee.jpg"]},
                                                                       {"text": "How to make this pastry?", "files":["./baklava.png"]}], 
                         description="Try [LLaVA Llama-3-8B](https://huggingface.co/xtuner/llava-llama-3-8b-v1_1-transformers). Upload an image and start chatting about it, or simply try one of the examples below. If you don't upload an image, you will receive an error.",
                         stop_btn="Stop Generation", multimodal=True)