Hugging Face's logo

Spaces:
akhaliq
/
Magistral-Small-2509
Runtime error

App Files Community
1
akhaliq HF Staff commited on
Commit
ae227b7
·
verified ·
1 Parent(s): d924f74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ model = Mistral3ForConditionalGeneration.from_pretrained(
33
 
34
  SYSTEM_PROMPT = load_system_prompt(model_id, "SYSTEM_PROMPT.txt")
35
 
36
- @spaces.zero_gpu(duration=120)
37
  def predict(message: str, image) -> str:
38
  messages = [
39
  SYSTEM_PROMPT,
 
33
 
34
  SYSTEM_PROMPT = load_system_prompt(model_id, "SYSTEM_PROMPT.txt")
35
 
36
+ @spaces.GPU(duration=120)
37
  def predict(message: str, image) -> str:
38
  messages = [
39
  SYSTEM_PROMPT,