use

README.md

@@ -1,6 +1,6 @@
 ---
-title: GLM-Edge-1.5B-Chat Space
-emoji: 🐨
+title: GLM-Edge-V-5B Space
+emoji: 📷
 colorFrom: green
 colorTo: indigo
 sdk: gradio
@@ -9,9 +9,9 @@ app_file: app.py
 pinned: false
 ---
 
-## GLM-Edge-1.5B-Chat Space
+## GLM-Edge-V-5B Space
 
-Run with Gradio on GLM-Edge-1.5B-Chat Space.
+Run with Gradio on GLM-Edge-V-5B Space.
 ```shell
 python app.py
 ```

app.py

@@ -5,6 +5,7 @@ from PIL import Image
 import re
 import gradio as gr
 import torch
+import spaces
 from transformers import (
     AutoTokenizer,
     AutoModelForCausalLM,
@@ -51,6 +52,7 @@ def preprocess_messages(history, image):
             print("Invalid image path. Continuing with text conversation.")
     return messages, pixel_values
 
+@spaces.GPU()
 def predict(history, max_length, top_p, temperature, image=None):
     messages, pixel_values = preprocess_messages(history, image)