Spaces:

jeydipak
/

nlp_project

Runtime error

jeydipak commited on Mar 8

Commit

4235e44

•

1 Parent(s): 0d10780

added torch

Files changed (1) hide show

pages/task3.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import streamlit as st
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import time
 def generate_text(model, tokenizer, prompt, max_length, num_generations, temperature):
     generated_texts = []

 import streamlit as st
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import time
+import torch
 def generate_text(model, tokenizer, prompt, max_length, num_generations, temperature):
     generated_texts = []