jon-fernandes commited on
Commit
fa5a060
·
verified ·
1 Parent(s): 7568d19

Uploading FoodExtract demo app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,14 +3,14 @@
3
  import time
4
  import transformers
5
  import torch
6
- import spaces # Optional: run our model on the GPU (this will be much faster inference)
7
 
8
  import gradio as gr
9
 
10
  from transformers import AutoModelForCausalLM, AutoTokenizer
11
  from transformers import pipeline
12
 
13
- @spaces.GPU # Optional: run our model on the GPU (this will be much faster inference)
14
  def pred_on_text(input_text):
15
  start_time = time.time()
16
 
 
3
  import time
4
  import transformers
5
  import torch
6
+ import spaces
7
 
8
  import gradio as gr
9
 
10
  from transformers import AutoModelForCausalLM, AutoTokenizer
11
  from transformers import pipeline
12
 
13
+ @spaces.GPU # Optional: run our model on the GPU
14
  def pred_on_text(input_text):
15
  start_time = time.time()
16