Update app.py

app.py

@@ -1,19 +1,13 @@
-import gradio as gr
-from transformers import AutoProcessor, AutoModelForCausalLM
+from PIL import Image
 import spaces
-
-import requests
-import copy
-
-from PIL import Image, ImageDraw, ImageFont 
-import io
-import matplotlib.pyplot as plt
-import matplotlib.patches as patches
-
-import random
-import numpy as np
-
+import gradio as gr
+from transformers import (
+    AutoProcessor,
+    AutoModelForCausalLM,
+)
+import torch
 import subprocess
+
 subprocess.run(
     "pip install flash-attn --no-build-isolation",
     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},