Update app.py

app.py

@@ -21,7 +21,7 @@ pipe = pipeline(
 
 # Text generation function
 def generate_text(prompt):
-    pipe.to('cuda')
+    pipe.model.to('cuda')
     response = pipe(prompt, max_length=100, num_return_sequences=1)
     return response[0]['generated_text']