zhengr commited on
Commit
594c06a
·
verified ·
1 Parent(s): 324a200

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
  from transformers import AutoModel, AutoTokenizer, AutoModelForCausalLM, TextStreamer
3
  from peft import PeftModel
4
  import re
@@ -9,7 +10,6 @@ import random
9
  import json
10
  import requests
11
  import math
12
- import spaces
13
  import torch
14
  from datetime import date
15
  from datetime import date, datetime, timedelta
@@ -390,7 +390,7 @@ def get_all_prompts_online(symbol, with_basics=True, max_news_perweek = 5, weeks
390
 
391
  return info, prompt
392
 
393
- @spaces.GPU
394
  def ask(symbol, weeks_before, withbasic):
395
 
396
  # load inference data
 
1
  import gradio as gr
2
+ import spaces
3
  from transformers import AutoModel, AutoTokenizer, AutoModelForCausalLM, TextStreamer
4
  from peft import PeftModel
5
  import re
 
10
  import json
11
  import requests
12
  import math
 
13
  import torch
14
  from datetime import date
15
  from datetime import date, datetime, timedelta
 
390
 
391
  return info, prompt
392
 
393
+ #@spaces.GPU
394
  def ask(symbol, weeks_before, withbasic):
395
 
396
  # load inference data