Update app.py

app.py

@@ -29,10 +29,15 @@ model = load_model_from_huggingface()
 # Force model to stay in eval mode
 model.train(False)
 
-def generate_text(prompt, max_length=100, num_samples=1):
+def generate_text(prompt, max_length="25", num_samples="1"):
     enc = tiktoken.get_encoding('gpt2')
     tokens = enc.encode(prompt)
     tokens = torch.tensor(tokens, dtype=torch.long)
+    
+    # Convert inputs to integers
+    max_length = int(max_length)
+    num_samples = int(num_samples)
+    
     tokens = tokens.unsqueeze(0).repeat(num_samples, 1)
     tokens = tokens.to(device)