ManishThota commited on
Commit
7f587f0
1 Parent(s): c935ba3

Update src/text_processor.py

Browse files
Files changed (1) hide show
  1. src/text_processor.py +1 -1
src/text_processor.py CHANGED
@@ -38,7 +38,7 @@ def load_model_pipeline(model_path: str):
38
  pipe = load_model_pipeline(model_path)
39
 
40
  # Generate output from LLM
41
- @spaces.GPU(duration=50)
42
  def generate_logic(llm_output: str) -> str:
43
  prompt = f"""
44
  Provide a detailed response based on the description: '{llm_output}'.
 
38
  pipe = load_model_pipeline(model_path)
39
 
40
  # Generate output from LLM
41
+ @spaces.GPU(duration=40)
42
  def generate_logic(llm_output: str) -> str:
43
  prompt = f"""
44
  Provide a detailed response based on the description: '{llm_output}'.