Rename app (2).py to app.py

app (2).py → app.py

@@ -31,12 +31,12 @@ from transformers import pipeline
 pipe = pipeline("text-generation",
                 model=model,
                 tokenizer= tokenizer,
-                torch_dtype=torch.float32,
+                torch_dtype=torch.float16,
                 device_map="auto",
                 max_new_tokens = 512,
                 do_sample=True,
                 top_k=30,
-                num_return_sequences=2,
+                num_return_sequences=1,
                 eos_token_id=tokenizer.eos_token_id
                 )