hallisky commited on
Commit
593928c
·
1 Parent(s): b77dbf2

Update temp file

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,9 +3,12 @@ import gradio as gr
3
  import torch
4
  from transformers import AutoTokenizer
5
 
 
 
6
  @spaces.GPU
7
  def temp(text):
8
- print(text)
 
9
 
10
  def greet(input_text, length, function_words, grade_level, sarcasm, formality, voice, persuasive, descriptive, narrative, expository):
11
  response = (
 
3
  import torch
4
  from transformers import AutoTokenizer
5
 
6
+ tokenizer = AutoTokenizer.from_pretrained("gpt2")
7
+
8
  @spaces.GPU
9
  def temp(text):
10
+ response = tokenizer(input_text, return_tensors="pt")
11
+ return response
12
 
13
  def greet(input_text, length, function_words, grade_level, sarcasm, formality, voice, persuasive, descriptive, narrative, expository):
14
  response = (