Update app.py

app.py

@@ -1,10 +1,11 @@
+import torch
 import gradio as gr
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
 def generate_text(input_text):
     # Load pre-trained model and tokenizer
     model_name = 'kurry/t5_small_finetuned'
-    model = T5ForConditionalGeneration.from_pretrained(model_name)
+    model = T5ForConditionalGeneration.from_pretrained(model_name, map_location=torch.device('cpu'))
     tokenizer = T5Tokenizer.from_pretrained(model_name)
 
     # Generate summary