CreitinGameplays commited on
Commit
a487b7f
1 Parent(s): 8ce6d50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
  model.to(device)
14
 
15
- @spaces.GPU(duration=120)
16
  def generate_text(user_prompt):
17
  """Generates text using the BLOOM model from Hugging Face Transformers and removes the user prompt."""
18
  # Construct the full prompt with system introduction, user prompt, and assistant role
 
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
  model.to(device)
14
 
15
+ @spaces.GPU(duration=60)
16
  def generate_text(user_prompt):
17
  """Generates text using the BLOOM model from Hugging Face Transformers and removes the user prompt."""
18
  # Construct the full prompt with system introduction, user prompt, and assistant role