Update app.py

app.py

@@ -1,15 +1,10 @@
 import spaces
-print('spaces.....................................')
 import os
 import argparse
 from functools import partial
 
 import torch
 import random
-
-# os.system('export CUDA_HOME=$(dirname $(dirname $(which nvcc)))')
-# os.system('pip install -r running_requirements.txt')
-
 import gradio as gr
 from src import MindOmni
 
@@ -38,7 +33,7 @@ def build_model(args):
     return MindOmni_model
 
 
-@spaces.GPU(duration=180)
+@spaces.GPU
 def understand_func(
         MindOmni_model, text, do_sample, temperature,
         max_new_tokens, input_llm_images):
@@ -50,7 +45,7 @@ def understand_func(
     return answer
 
 
-@spaces.GPU(duration=180)
+@spaces.GPU
 def generate_func(
         MindOmni_model, text, use_cot, height, width, guidance_scale, inference_steps, seed, separate_cfg_infer, offload_model, max_input_image_size, randomize_seed, save_images, do_sample, temperature, max_new_tokens, input_llm_images, only_understand):
     if input_llm_images is not None and not isinstance(input_llm_images, list):
@@ -59,14 +54,12 @@ def generate_func(
     if randomize_seed:
         seed = random.randint(0, 10000000)
 
-    os.makedirs(os.path.dirname('/tmp/.unhold'), exist_ok=True)
-    with open('/tmp/.unhold', 'w') as f:
-        f.write('')
+    print(f'Generate image prompt: {text}')
     output, prompt_ = MindOmni_model.generate_image(
         height, width, guidance_scale, inference_steps, separate_cfg_infer, offload_model, seed, max_input_image_size,
         text, NEGATIVE_PROMPT, input_llm_images, do_sample, temperature, max_new_tokens, only_understand, use_cot=use_cot)
-    os.remove('/tmp/.unhold')
-
+    print('Generation finished.')
+    
     img = output[0]
 
     if save_images: