Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gdown
 import torch
 import torch.nn as nn
@@ -14,7 +15,9 @@ itos = {i: ch for i, ch in enumerate(chars)}
 
 # Load the model
 gdown.download('https://drive.usercontent.google.com/download?id=14k2xUrvJ32trhLCzV2_O7klreBBA3dUu&authuser=0&confirm=t', 'model.pth', quiet=False)
-model = torch.load('model.pth')
+
+# Load the model with CPU mapping
+model = torch.load('model.pth', map_location=torch.device('cpu'))
 
 # Function for generating text
 def generate_text(prompt, max_length=50):
@@ -33,4 +36,3 @@ if __name__ == "__main__":
     prompt = "Enter your prompt here"  # Replace with actual input
     generated = generate_text(prompt)
     print(generated)
-