Update app.py

app.py

@@ -13,7 +13,7 @@ generator = pipeline('text-generation', model='mistralai/Mistral-7B-v0.1', torch
 def generate_text(prompt, temperature, top_p, top_k, repetition_penalty, max_length):
     # Generate text using the model
     generator.model.cuda()
-    generator.device = "cuda"
+    generator.device = torch.device("cuda")
     outputs = generator(
         prompt, 
         max_new_tokens=max_length,