akhaliq HF staff commited on
Commit
1468560
1 Parent(s): 66a6386

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -14,6 +14,7 @@ from transformers.generation import (
14
  )
15
  import torch
16
  from emu3.mllm.processing_emu3 import Emu3Processor
 
17
 
18
  # Model paths
19
  EMU_GEN_HUB = "BAAI/Emu3-Gen"
@@ -59,6 +60,7 @@ chat_processor = Emu3Processor(
59
  chat_image_processor, chat_image_tokenizer, chat_tokenizer
60
  )
61
 
 
62
  def generate_image(prompt):
63
  POSITIVE_PROMPT = " masterpiece, film grained, best quality."
64
  NEGATIVE_PROMPT = (
 
14
  )
15
  import torch
16
  from emu3.mllm.processing_emu3 import Emu3Processor
17
+ import spaces
18
 
19
  # Model paths
20
  EMU_GEN_HUB = "BAAI/Emu3-Gen"
 
60
  chat_image_processor, chat_image_tokenizer, chat_tokenizer
61
  )
62
 
63
+ @spaces.GPU(duration=120)
64
  def generate_image(prompt):
65
  POSITIVE_PROMPT = " masterpiece, film grained, best quality."
66
  NEGATIVE_PROMPT = (